From edab0ea0a6f6fc0289a93b391372f3451588b24e Mon Sep 17 00:00:00 2001 From: George Hotz Date: Mon, 13 Sep 2021 17:35:05 -0700 Subject: [PATCH] fix param setting old-commit-hash: 9b610726902c9713ec63c304c59ca3dd5f708449 --- common/params.py | 17 ++++++----------- 1 file changed, 6 insertions(+), 11 deletions(-) diff --git a/common/params.py b/common/params.py index d0a37cce1..2cfca3712 100644 --- a/common/params.py +++ b/common/params.py @@ -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)}")