Skip to content

Commit

Permalink
fix hybrid
Browse files Browse the repository at this point in the history
Signed-off-by: Jinzhe Zeng <[email protected]>
  • Loading branch information
njzjz committed Nov 7, 2024
1 parent 1686ff1 commit 9605e9a
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions deepmd/dpmodel/descriptor/hybrid.py
Original file line number Diff line number Diff line change
Expand Up @@ -58,9 +58,10 @@ def __init__(
if isinstance(ii, BaseDescriptor):
formatted_descript_list.append(ii)
elif isinstance(ii, dict):
formatted_descript_list.append(
BaseDescriptor(**ii, type_map=type_map, ntypes=ntypes)
)
ii = ii.copy()
ii["type_map"] = type_map
ii["ntypes"] = ntypes
formatted_descript_list.append(BaseDescriptor(**ii))
else:
raise NotImplementedError
self.descrpt_list = formatted_descript_list
Expand Down

0 comments on commit 9605e9a

Please sign in to comment.