ANL-CEEESA/MIPLearn
1
0
Fork 0
mirror of https://github.com/ANL-CEEESA/MIPLearn.git synced 2025-12-06 01:18:52 -06:00
Code Issues Packages Projects Releases Wiki Activity

Add progress arg to LearningSolver.solve

Browse Source
This commit is contained in:
Alinson S. Xavier
2022-02-25 09:35:05 -06:00
parent ce78d5114a
commit f794c27634
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
miplearn/solvers/learning.py
View File

@@ -379,11 +379,12 @@ class LearningSolver:
arg: Union[Any, List[str]], arg: Union[Any, List[str]],
build_model: Optional[Callable] = None, build_model: Optional[Callable] = None,
tee: bool = False, tee: bool = False,
progress: bool = False,
) -> Union[LearningSolveStats, List[LearningSolveStats]]: ) -> Union[LearningSolveStats, List[LearningSolveStats]]:
if isinstance(arg, list): if isinstance(arg, list):
assert build_model is not None assert build_model is not None
stats = [] stats = []
for i in arg: for i in tqdm(arg, disable=not progress):
s = self._solve(FileInstanceWrapper(i, build_model), tee=tee) s = self._solve(FileInstanceWrapper(i, build_model), tee=tee)
stats.append(s) stats.append(s)
return stats return stats