fix param setting

pull/22222/head
George Hotz 2021-09-13 17:35:05 -07:00
parent de478b75fb
commit 9b61072690
1 changed files with 6 additions and 11 deletions

View File

@ -6,19 +6,14 @@ assert put_nonblocking
if __name__ == "__main__":
import sys
from common.params_pyx import keys # pylint: disable=no-name-in-module, import-error
params = Params()
key = sys.argv[1]
assert params.check_key(key), f"unknown param: {key}"
if len(sys.argv) == 3:
name = sys.argv[1]
val = sys.argv[2]
assert name.encode("utf-8") in keys.keys(), f"unknown param: {name}"
print(f"SET: {name} = {val}")
params.put(name, val)
print(f"SET: {key} = {val}")
params.put(key, val)
elif len(sys.argv) == 2:
name = sys.argv[1]
assert name.encode("utf-8") in keys.keys(), f"unknown param: {name}"
print(f"GET: {name} = {params.get(name)}")
else:
for k in keys.keys():
print(f"GET: {k} = {params.get(k)}")
print(f"GET: {key} = {params.get(key)}")