fix ut

caic99 · caic99 · commit 7a7eb60e35c0 · 2025-06-18T03:26:31.000Z
diff --git a/deepmd/dpmodel/utils/env_mat_stat.py b/deepmd/dpmodel/utils/env_mat_stat.py
@@ -166,7 +166,7 @@ def iter(
                     self.last_dim,
                 ),
             )
-            atype = xp.reshape(atype, (coord.shape[0] * coord.shape[1]))
+            atype = xp.reshape(atype, (coord.shape[0] * coord.shape[1],))
             # (1, nloc) eq (ntypes, 1), so broadcast is possible
             # shape: (ntypes, nloc)
             type_idx = xp.equal(

Original file line number	Diff line number	Diff line change
`@@ -166,7 +166,7 @@ def iter(`
`166`	`166`	`self.last_dim,`
`167`	`167`	`),`
`168`	`168`	`)`
`169`		`- atype = xp.reshape(atype, (coord.shape[0] * coord.shape[1]))`
	`169`	`+ atype = xp.reshape(atype, (coord.shape[0] * coord.shape[1],))`
`170`	`170`	`# (1, nloc) eq (ntypes, 1), so broadcast is possible`
`171`	`171`	`# shape: (ntypes, nloc)`
`172`	`172`	`type_idx = xp.equal(`