set solver in benchmark

cdanielmachado · Apr 25, 2023 · b55cd63 · b55cd63
1 parent 09c2f31
commit b55cd63
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/carveme/cli/benchmark.py b/carveme/cli/benchmark.py
@@ -1,5 +1,5 @@
 import pandas as pd
-from reframed import load_cbmodel
+from reframed import load_cbmodel, set_default_solver
 from carveme.reconstruction.utils import load_media_db
 from carveme.reconstruction.benchmark import benchmark_biolog, benchmark_essentiality, mcc
 from carveme import project_dir
@@ -249,8 +249,13 @@ def main():
 
     parser.add_argument('--species', dest='species', help="Benchmark only one species.")
 
+    parser.add_argument('--solver', help="Select MILP solver. Available options: cplex [default], gurobi.")
+
     args = parser.parse_args()
 
+    if args.solver:
+        set_default_solver(args.solver)
+
     benchmark(rebuild=(not args.no_rebuild),
               biolog=(not args.no_biolog),
               essentiality=(not args.no_essentiality),