Update run.py

deepmodeling · May 9, 2024 · a500987 · a500987
1 parent 083ad45
commit a500987
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'"
+    # assert train_command == "dp", "The 'train_command' should be 'dp'"  # should modify tests for this command
     if suffix == ".pb":
         train_command += " --tf"
     elif suffix == ".pth":