Raise if cupy is installed but cuda isn't

loganbvh · Jan 30, 2024 · d52ceac · d52ceac
1 parent 3c7d3f7
commit d52ceac
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tdgl/finite_volume/operators.py b/tdgl/finite_volume/operators.py
@@ -6,10 +6,11 @@
 
 try:
     import cupy  # type: ignore
-    from cupyx.scipy.sparse import csc_matrix, csr_matrix  # type: ignore
-    from cupyx.scipy.sparse.linalg import factorized  # type: ignore
 except ImportError:
     cupy = None
+else:
+    from cupyx.scipy.sparse import csc_matrix, csr_matrix  # type: ignore
+    from cupyx.scipy.sparse.linalg import factorized  # type: ignore
 
 from ..solver.options import SparseSolver
 from .mesh import Mesh