use K from name

commaai · Dec 3, 2020 · 2f05431 · 2f05431
1 parent 8de9301
commit 2f05431
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/rednose/helpers/feature_handler.py b/rednose/helpers/feature_handler.py
@@ -153,6 +153,6 @@ def generate_orient_error_jac(K):
 
 
 if __name__ == "__main__":
-  # TODO: get K from argparse
+  K = int(sys.argv[1].split("_")[-1])
   generated_dir = sys.argv[2]
-  FeatureHandler.generate_code(generated_dir)
+  FeatureHandler.generate_code(generated_dir, K=K)