use syntax compatible with python 3.10

Signed-off-by: Victor Garcia Reolid <[email protected]>

Author: Victor Garcia Reolid

pytensor/link/numba/dispatch/basic.py

@@ -476,7 +476,9 @@ def argort_vec(X, axis):
 
             for i in prange(N):
                 idx = indices[i]
-                result[:, *idx] = np.argsort(Y[:, *idx], kind=kind)
+                result[(slice(None), *idx)] = np.argsort(
+                    Y[(slice(None), *idx)], kind=kind
+                )
 
             result = np.swapaxes(result, 0, axis)