Add output files to missing CLI functions

ElliottKasoar · Feb 27, 2025 · 4e13369 · 4e13369
1 parent 8d3c8e7
commit 4e13369
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 6 deletions.
diff --git a/janus_core/cli/preprocess.py b/janus_core/cli/preprocess.py
@@ -51,13 +51,17 @@ def preprocess(
 
     inputs = {"mlip_config": str(mlip_config)}
 
-    # Save summary information before preprocessing begins
-    start_summary(command="preprocess", summary=summary, inputs=inputs)
-
     log_kwargs = {"filemode": "w"}
     if log:
         log_kwargs["filename"] = log
 
+    output_files = {"log": log}
+
+    # Save summary information before preprocessing begins
+    start_summary(
+        command="preprocess", summary=summary, inputs=inputs, output_files=output_files
+    )
+
     # Run preprocessing
     run_preprocess(
         mlip_config, attach_logger=True, log_kwargs=log_kwargs, track_carbon=tracker

diff --git a/janus_core/cli/train.py b/janus_core/cli/train.py
@@ -77,13 +77,17 @@ def train(
 
     inputs = {"mlip_config": str(mlip_config), "fine_tune": fine_tune}
 
-    # Save summary information before training begins
-    start_summary(command="train", summary=summary, inputs=inputs)
-
     log_kwargs = {"filemode": "w"}
     if log:
         log_kwargs["filename"] = log
 
+    output_files = {"log": log}
+
+    # Save summary information before training begins
+    start_summary(
+        command="train", summary=summary, inputs=inputs, output_files=output_files
+    )
+
     # Run training
     run_train(
         mlip_config, attach_logger=True, log_kwargs=log_kwargs, track_carbon=tracker