[trainer_llm] 0.0.65 Fix arguments parsing

TCLResearchEurope · May 20, 2024 · 718801f · 718801f
1 parent 4daaff7
commit 718801f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/examples/trainer_llm/run.py b/examples/trainer_llm/run.py
@@ -55,6 +55,9 @@ def parse_args() -> argparse.Namespace:
 
     arg_parser = argparse.ArgumentParser()
     arg_parser.add_argument("--version", action="store_true")
+    arg_parser.add_argument("--output-path", type=pathlib.Path)
+    arg_parser.add_argument("--config", type=pathlib.Path)
+
     args = arg_parser.parse_args()
 
     # If no --version, run parsing of trainign/decomposition arguments
@@ -72,6 +75,7 @@ def parse_args() -> argparse.Namespace:
             required=True,
         )
         args = arg_parser.parse_args()
+        args.version = False
 
     return args
 

diff --git a/examples/trainer_llm/version.py b/examples/trainer_llm/version.py
@@ -1,5 +1,5 @@
 # This is parsed by setup.py, so we need to stick to str -> int parsing
-__version__ = "0.0.64"
+__version__ = "0.0.65"
 
 _ver_major = int(__version__.split(".")[0])
 _ver_minor = int(__version__.split(".")[1])