Add progress arg to LearningSolver.solve

master
Alinson S. Xavier 4 years ago
parent ce78d5114a
commit f794c27634
No known key found for this signature in database
GPG Key ID: DCA0DAD4D2F58624

@ -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

Loading…
Cancel
Save