fix config-path arg bug in CLIs

mllam · Nov 13, 2024 · a95eb5a · a95eb5a
1 parent 89fac82
commit a95eb5a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/neural_lam/create_graph.py b/neural_lam/create_graph.py
@@ -590,7 +590,9 @@ def cli(input_args=None):
     )
     args = parser.parse_args(input_args)
 
-    assert args.config is not None, "Specify your config with --config_path"
+    assert (
+        args.config_path is not None
+    ), "Specify your config with --config_path"
 
     # Load neural-lam configuration and datastore to use
     _, datastore = load_config_and_datastore(config_path=args.config_path)

diff --git a/neural_lam/train_model.py b/neural_lam/train_model.py
@@ -209,7 +209,9 @@ def main(input_args=None):
     }
 
     # Asserts for arguments
-    assert args.config is not None, "Specify your config with --config_path"
+    assert (
+        args.config_path is not None
+    ), "Specify your config with --config_path"
     assert args.model in MODELS, f"Unknown model: {args.model}"
     assert args.eval in (
         None,