Update run.py

deepmodeling · May 9, 2024 · 35b2713 · 35b2713
1 parent a500987
commit 35b2713
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dpgen/generator/run.py b/dpgen/generator/run.py
@@ -749,7 +749,7 @@ def run_train(iter_index, jdata, mdata):
         )
 
     train_command = mdata.get("train_command", "dp").strip()
-    # assert train_command == "dp", "The 'train_command' should be 'dp'"  # should modify tests for this command
+    # assert train_command == "dp", "The 'train_command' should be 'dp'"     # the tests should be updated to run this command
     if suffix == ".pb":
         train_command += " --tf"
     elif suffix == ".pth":