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__": if __name__ == "__main__":
import sys import sys
from common.params_pyx import keys # pylint: disable=no-name-in-module, import-error
params = Params() params = Params()
key = sys.argv[1]
assert params.check_key(key), f"unknown param: {key}"
if len(sys.argv) == 3: if len(sys.argv) == 3:
name = sys.argv[1]
val = sys.argv[2] val = sys.argv[2]
assert name.encode("utf-8") in keys.keys(), f"unknown param: {name}" print(f"SET: {key} = {val}")
print(f"SET: {name} = {val}") params.put(key, val)
params.put(name, val)
elif len(sys.argv) == 2: elif len(sys.argv) == 2:
name = sys.argv[1] print(f"GET: {key} = {params.get(key)}")
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)}")