fix: update shape assignment in xp_take_along_axis to use tuple unpacking

caic99 · caic99 · commit ca9ae69af540 · 2025-06-17T09:15:02.000Z
diff --git a/deepmd/dpmodel/array_api.py b/deepmd/dpmodel/array_api.py
@@ -60,7 +60,7 @@ def xp_take_along_axis(arr, indices, axis):
 
     shape = list(arr.shape)
     shape.pop(-1)
-    shape = [*shape, n]
+    shape = (*shape, n)
 
     arr = xp.reshape(arr, (-1,))
     if n != 0:
