Backport PR scverse#1754: (fix): resolve cupy import error on `cupy-c…

…uda12x<0.13`
meeseeksmachine · Nov 11, 2024 · 21d0ff7 · 21d0ff7
1 parent cb8b760
commit 21d0ff7
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/docs/release-notes/1754.bugfix.md b/docs/release-notes/1754.bugfix.md
@@ -0,0 +1 @@
+Fix `cupy<0.13` imports in non-gpu environments {user}`ilan-gold`
diff --git a/src/anndata/compat/__init__.py b/src/anndata/compat/__init__.py
@@ -140,7 +140,16 @@ def __repr__():
             return "mock dask.array.core.Array"
 
 
-if find_spec("cupy") or TYPE_CHECKING:
+# https://github.com/scverse/anndata/issues/1749
+def is_cupy_importable() -> bool:
+    try:
+        import cupy  # noqa: F401
+    except ImportError:
+        return False
+    return True
+
+
+if is_cupy_importable() or TYPE_CHECKING:
     from cupy import ndarray as CupyArray
     from cupyx.scipy.sparse import csc_matrix as CupyCSCMatrix
     from cupyx.scipy.sparse import csr_matrix as CupyCSRMatrix