Merge pull request #84 from sbintuitions/feature/save_config

[Feature] update save config file
sbintuitions · Nov 21, 2024 · ccadd5d · ccadd5d
2 parents 3a08ca8 + c10189f
commit ccadd5d
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/src/jmteb/__main__.py b/src/jmteb/__main__.py
@@ -97,6 +97,16 @@ def main(
     if len(args.evaluators) == 0:
         raise ValueError("No evaluator is selected. Please check the config file or the command line arguments.")
 
+    # save config as yaml
+    parser.save(
+        args,
+        Path(args.save_dir) / "jmteb_config.yaml",
+        format="yaml",
+        overwrite=True,
+        multifile=False,
+        skip_check=True,
+    )
+
     args = parser.instantiate_classes(args)
     if isinstance(args.evaluators, str):
         raise ValueError(