fix pyright

wanghan-iapcm · Jun 5, 2024 · c7bf8fc · c7bf8fc
1 parent 2e1e993
commit c7bf8fc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dpdata/system.py b/dpdata/system.py
@@ -776,7 +776,7 @@ def replicate(self, ncopy: list[int] | tuple[int, int, int]):
         tmp.data["atom_numbs"] = list(
             np.array(np.copy(data["atom_numbs"])) * np.prod(ncopy)
         )
-        tmp.data["atom_types"] = np.tile(np.copy(data["atom_types"]), (np.prod(ncopy),) + (1,))
+        tmp.data["atom_types"] = np.tile(np.copy(data["atom_types"]), (int(np.prod(ncopy)),) + (1,))
         tmp.data["atom_types"] = np.transpose(tmp.data["atom_types"]).reshape([-1])
 
         tmp.data["cells"] = np.copy(data["cells"])