cell may be none

njzjz · Sep 24, 2024 · be6a8dc · be6a8dc
1 parent b3f0dd7
commit be6a8dc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepmd/dpmodel/utils/nlist.py b/deepmd/dpmodel/utils/nlist.py
@@ -273,7 +273,7 @@ def extend_coord_with_ghosts(
         maping extended index to the local index
 
     """
-    xp = array_api_compat.array_namespace(coord, atype, cell)
+    xp = array_api_compat.array_namespace(coord, atype)
     nf, nloc = atype.shape
     aidx = xp.tile(xp.arange(nloc)[xp.newaxis, :], (nf, 1))  # pylint: disable=no-explicit-dtype
     if cell is None: