fix: natom error in calc_model_devi (#3597)

#3596
deepmodeling · Mar 26, 2024 · 3c09d46 · 3c09d46
1 parent 98f69e0
commit 3c09d46
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepmd/infer/model_devi.py b/deepmd/infer/model_devi.py
@@ -303,7 +303,7 @@ def calc_model_devi(
     energies = []
     forces = []
     virials = []
-    natom = atype.shape[-1]
+    natom = np.array(atype).shape[-1]
     for dp in models:
         ret = dp.eval(
             coord,