fix jit error

Signed-off-by: Jinzhe Zeng <[email protected]>
dptech-corp · Jan 25, 2024 · 6cb4b6c · 6cb4b6c
1 parent 0bbe17d
commit 6cb4b6c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepmd_pt/utils/region.py b/deepmd_pt/utils/region.py
@@ -65,7 +65,7 @@ def to_face_distance(
     """
     cshape = cell.shape
     dist = b_to_face_distance(cell.view([-1, 3, 3]))
-    return dist.view([*list(cshape[:-2]), 3])
+    return dist.view(list(cshape[:-2]) + [3])  # noqa:RUF005
 
 
 def _to_face_distance(cell):