Skip to content

Commit

Permalink
fix: fix typo in run_train (#1581)
Browse files Browse the repository at this point in the history
<!-- This is an auto-generated comment: release notes by coderabbit.ai
-->

## Summary by CodeRabbit

- **Refactor**
- Renamed internal function `make_train_dp` to `run_train_dp` for better
clarity and consistency within the codebase.

(Note: This change does not impact end-user functionalities directly.)

<!-- end of auto-generated comment: release notes by coderabbit.ai -->

Signed-off-by: Jinzhe Zeng <[email protected]>
  • Loading branch information
njzjz authored Jul 10, 2024
1 parent 2f39d12 commit 30bc1e5
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion dpgen/generator/run.py
Original file line number Diff line number Diff line change
Expand Up @@ -728,7 +728,7 @@ def get_nframes(system):
def run_train(iter_index, jdata, mdata):
mlp_engine = jdata.get("mlp_engine", "dp")
if mlp_engine == "dp":
return make_train_dp(iter_index, jdata, mdata)
return run_train_dp(iter_index, jdata, mdata)
else:
raise ValueError(f"Unsupported engine: {mlp_engine}")

Expand Down

0 comments on commit 30bc1e5

Please sign in to comment.