fix TypeError when type_map is not given

Fix #2889. Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · Oct 2, 2023 · 2b90ebe · 2b90ebe
1 parent cf61140
commit 2b90ebe
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepmd/model/model.py b/deepmd/model/model.py
@@ -531,7 +531,7 @@ def __init__(
             self.descrpt = descriptor
         else:
             self.descrpt = Descriptor(
-                **descriptor, ntypes=len(type_map), spin=self.spin
+                **descriptor, ntypes=len(self.type_map), spin=self.spin
             )
 
         if isinstance(fitting_net, Fitting):