Update generation script to use COIN & MILP if needed

AntaresSimulatorTeam · Dec 8, 2023 · d0e0e2f · d0e0e2f
1 parent c058e3b
commit d0e0e2f
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/scripts/generate_reference.py b/scripts/generate_reference.py
@@ -38,10 +38,21 @@ def find_solver(solver):
 
 studies = antares_utils.list_studies(root)
 
+def solver_config(study_name):
+    if study_name == "valid-milp":
+        return ("coin", True)
+    else:
+        return ("sirius", False)
+
 for study in studies:
     print(study.name + '...', end='')
+
+    # Do we need named MPS problems ?
     named_mps_problems = (study.parent.name == 'valid-named-mps')
+    # What optimization solver to use ?
+    (opt_solver, use_ortools) = solver_config(study.parent.name)
+
+    result = antares_utils.generate_reference_values(solver_path, study, use_ortools, opt_solver, named_mps_problems)
 
-    result = antares_utils.generate_reference_values(solver_path, study, False, "sirius", named_mps_problems)
 
     print('OK' if result else 'KO')