Infer solvers that require sparse inputs

ZAKIAkram · Apr 3, 2023 · 3b994dd · 3b994dd
1 parent ee6634c
commit 3b994dd
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/qpsolvers_benchmark/utils.py b/qpsolvers_benchmark/utils.py
@@ -117,7 +117,10 @@ def time_solve_problem(
         to compute it.
     """
     # Don't time matrix conversions for solvers that require sparse inputs
-    if solver in ["highs", "osqp", "scs"]:
+    if (
+        solver in qpsolvers.sparse_solvers
+        and solver not in qpsolvers.dense_solvers
+    ):
         problem = problem.to_sparse()
     start_time = perf_counter()
     try: