fix output arg typo

dsethz · Nov 9, 2024 · 13920c4 · 13920c4
1 parent c934da5
commit 13920c4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/nuclai/models/evaluate_lr_rf.py b/src/nuclai/models/evaluate_lr_rf.py
@@ -56,7 +56,7 @@ def main():
     args = args_parse()
     path_data = args.data
     path_model = args.model
-    path_out = args.out
+    path_out = args.output_base_dir
 
     # create output directory
     os.makedirs(path_out, exist_ok=True)

diff --git a/src/nuclai/models/train_lr_rf.py b/src/nuclai/models/train_lr_rf.py
@@ -49,7 +49,7 @@ def args_parse():
 def main():
     args = args_parse()
     path_data = args.data
-    path_out = args.out
+    path_out = args.output_base_dir
 
     # create output directory
     os.makedirs(path_out, exist_ok=True)