Skip to content

Commit 44dc9c5

Browse files
authored
fix: fix typo in run_train
Signed-off-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
1 parent 2f39d12 commit 44dc9c5

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

dpgen/generator/run.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -728,7 +728,7 @@ def get_nframes(system):
728728
def run_train(iter_index, jdata, mdata):
729729
mlp_engine = jdata.get("mlp_engine", "dp")
730730
if mlp_engine == "dp":
731-
return make_train_dp(iter_index, jdata, mdata)
731+
return run_train_dp(iter_index, jdata, mdata)
732732
else:
733733
raise ValueError(f"Unsupported engine: {mlp_engine}")
734734

0 commit comments

Comments
 (0)