Update runner.py

loganbvh · Sep 23, 2023 · 0d45f3f · 0d45f3f
1 parent a9085e4
commit 0d45f3f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tdgl/solver/runner.py b/tdgl/solver/runner.py
@@ -14,7 +14,7 @@
 from tqdm import TqdmWarning, tqdm
 
 from ..finite_volume.mesh import Mesh
-from .options import SolverOptions, SparseSolver
+from .options import SolverOptions
 
 
 def _get(item):
@@ -233,7 +233,7 @@ def __init__(
         self.running_names_and_sizes = (
             running_names_and_sizes if running_names_and_sizes is not None else {}
         )
-        if options.sparse_solver is SparseSolver.CUPY:
+        if options.gpu:
             import cupy  # type: ignore
 
             array_module = cupy