data-apis · Nov 30, 2024 · Nov 30, 2024
Showing with 4 additions and 4 deletions.

+1 −1 pixi.lock

+1 −1 src/array_api_extra/__init__.py

+2 −2 src/array_api_extra/_lib/_utils.py
diff --git a/pixi.lock b/pixi.lock
diff --git a/src/array_api_extra/__init__.py b/src/array_api_extra/__init__.py
@@ -2,7 +2,7 @@
 
 from ._funcs import atleast_nd, cov, create_diagonal, expand_dims, kron, setdiff1d, sinc
 
-__version__ = "0.3.2"
+__version__ = "0.3.3"
 
 # pylint: disable=duplicate-code
 __all__ = [

diff --git a/src/array_api_extra/_lib/_utils.py b/src/array_api_extra/_lib/_utils.py
@@ -33,11 +33,11 @@ def in1d(
     # This code is run to make the code significantly faster
     if x2.shape[0] < 10 * x1.shape[0] ** 0.145:
         if invert:
-            mask = xp.ones(x1.shape[0], dtype=xp.bool, device=x1.device)
+            mask = xp.ones(x1.shape[0], dtype=xp.bool, device=_compat.device(x1))
             for a in x2:
                 mask &= x1 != a
         else:
-            mask = xp.zeros(x1.shape[0], dtype=xp.bool, device=x1.device)
+            mask = xp.zeros(x1.shape[0], dtype=xp.bool, device=_compat.device(x1))
             for a in x2:
                 mask |= x1 == a
         return mask