MAINT: reshape does not accept int

crusaderky · crusaderky · commit dce1a6f2c781 · 2025-05-16T11:02:22.000+01:00
diff --git a/src/array_api_extra/_lib/_funcs.py b/src/array_api_extra/_lib/_funcs.py
@@ -725,7 +725,8 @@ def nunique(x: Array, /, *, xp: ModuleType | None = None) -> Array:
         return xp.asarray(n, device=_compat.device(x))
 
     # xp does not have unique_counts; O(n*logn) complexity
-    x = xp.sort(xp.reshape(x, -1))
+    x = xp.reshape(x, (-1, ))
+    x = xp.sort(x)
     mask = x != xp.roll(x, -1)
     default_int = xp.__array_namespace_info__().default_dtypes(
         device=_compat.device(x)
