Fix dtype of numba dispatch of ArgSort

ricardoV94 · jessegrabowski · commit aa372e28ca8a · 2025-05-23T10:32:07.000+08:00
diff --git a/pytensor/link/numba/dispatch/basic.py b/pytensor/link/numba/dispatch/basic.py
@@ -466,7 +466,7 @@ def argort_vec(X, axis):
             axis = axis.item()
 
             Y = np.swapaxes(X, axis, 0)
-            result = np.empty_like(Y)
+            result = np.empty_like(Y, dtype="int64")
 
             indices = list(np.ndindex(Y.shape[1:]))
 
