From 35b27130e8e26c13bd0119d6970e5c8dd5d98275 Mon Sep 17 00:00:00 2001
From: Thang Nguyen <46436648+thangckt@users.noreply.github.com>
Date: Thu, 9 May 2024 14:39:49 +0900
Subject: [PATCH] Update run.py

---
 dpgen/generator/run.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/dpgen/generator/run.py b/dpgen/generator/run.py
index 69917eb6c..079723ef6 100644
--- 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":