MAINT: inline the check for complex dtypes

Preferred as long as xp.isdtype is not universally available (looking at you pytorch)
data-apis · Nov 20, 2024 · 6ea8ae2 · 6ea8ae2
1 parent fe3e05c
commit 6ea8ae2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/array_api_tests/test_linalg.py b/array_api_tests/test_linalg.py
@@ -907,8 +907,8 @@ def true_trace(x_stack, offset=0):
 
 
 def _conj(x):
-    """Work around xp.conj rejecting floats."""
-    if xp.isdtype(x.dtype, 'complex floating'):
+    # XXX: replace with xp.dtype when all array libraries implement it 
+    if x.dtype in (xp.complex64, xp.complex128):
         return xp.conj(x)
     else:
         return x