Skip to content

Commit 8c52a7d

Browse files
committed
eliminate code inconsistency raised by coderabbit
1 parent 24325c1 commit 8c52a7d

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

deepmd/pt/model/atomic_model/dp_atomic_model.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -154,7 +154,8 @@ def change_type_map(
154154
)
155155
self.fitting_net.change_type_map(type_map=type_map)
156156
# Reinitialize fitting to get correct sel_type
157-
self.fitting_net.reinit_exclude(self.atom_exclude_types)
157+
if hasattr(self.fitting_net, "reinit_exclude"):
158+
self.fitting_net.reinit_exclude(self.atom_exclude_types)
158159

159160
def has_message_passing(self) -> bool:
160161
"""Returns whether the atomic model has message passing."""

0 commit comments

Comments
 (0)