Skip to content

Commit e9a6bd4

Browse files
authored
chore(tf): throw an error if type map is missing in change_energy_bias (#4636)
Fix #4608. <!-- This is an auto-generated comment: release notes by coderabbit.ai --> ## Summary by CodeRabbit - **Bug Fixes** - Strengthened input validation to ensure that required configuration data is provided, resulting in a clearer notification when essential data is missing. <!-- end of auto-generated comment: release notes by coderabbit.ai --> Signed-off-by: Jinzhe Zeng <jinzhe.zeng@ustc.edu.cn>
1 parent e4eb01f commit e9a6bd4

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

deepmd/tf/train/trainer.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -892,6 +892,10 @@ def _change_energy_bias(
892892
bias_adjust_mode="change-by-statistic",
893893
) -> None:
894894
full_type_map = data.get_type_map()
895+
if len(full_type_map) == 0:
896+
raise ValueError(
897+
"The type_map.raw file must be provided in the input data."
898+
)
895899
self.model.change_energy_bias(
896900
data,
897901
frozen_model,

0 commit comments

Comments
 (0)