From f794c27634cbe3d89155a13222b115b7c4defcfd Mon Sep 17 00:00:00 2001
From: "Alinson S. Xavier" <git@axavier.org>
Date: Fri, 25 Feb 2022 09:35:05 -0600
Subject: [PATCH] Add progress arg to LearningSolver.solve

---
 miplearn/solvers/learning.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/miplearn/solvers/learning.py b/miplearn/solvers/learning.py
index 75e1748..90a7090 100644
--- a/miplearn/solvers/learning.py
+++ b/miplearn/solvers/learning.py
@@ -379,11 +379,12 @@ class LearningSolver:
         arg: Union[Any, List[str]],
         build_model: Optional[Callable] = None,
         tee: bool = False,
+        progress: bool = False,
     ) -> Union[LearningSolveStats, List[LearningSolveStats]]:
         if isinstance(arg, list):
             assert build_model is not None
             stats = []
-            for i in arg:
+            for i in tqdm(arg, disable=not progress):
                 s = self._solve(FileInstanceWrapper(i, build_model), tee=tee)
                 stats.append(s)
             return stats