From 0d45f3f995025194ce5eb0f7550158c749224739 Mon Sep 17 00:00:00 2001
From: Logan Bishop-Van Horn <logan.bvh@gmail.com>
Date: Sat, 23 Sep 2023 15:00:00 -0700
Subject: [PATCH] Update runner.py

---
 tdgl/solver/runner.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tdgl/solver/runner.py b/tdgl/solver/runner.py
index 8c07942..53645ba 100644
--- 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