From 718801f06288e005a99ff4eac7486f96371c30ad Mon Sep 17 00:00:00 2001
From: Michal Lopuszynski <michal.lopuszynski@tcl.com>
Date: Mon, 20 May 2024 22:21:44 +0200
Subject: [PATCH] [trainer_llm] 0.0.65 Fix arguments parsing

---
 examples/trainer_llm/run.py     | 4 ++++
 examples/trainer_llm/version.py | 2 +-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/examples/trainer_llm/run.py b/examples/trainer_llm/run.py
index ae22c46..80b1970 100644
--- 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
index ac904f4..dc0b11d 100644
--- 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])