Merge pull request #30 from TobyBoyne/master

Add verbosity parameter
cog-imperial · May 6, 2024 · 5d90ab3 · 5d90ab3
2 parents 7c4dbf8 + a2f3105
commit 5d90ab3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/entmoot/optimizers/pyomo_opt.py b/entmoot/optimizers/pyomo_opt.py
@@ -96,7 +96,8 @@ def solve(
         tree_model.add_to_pyomo_model(opt_model)
 
         # Solve optimization model
-        opt.solve(opt_model, tee=True)
+        verbose = self._params.get("verbose", True)
+        opt.solve(opt_model, tee=verbose)
 
         # update current solution
         self._curr_sol, self._active_leaves = self._get_sol(opt_model)