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

BasicCollector: Make log file optional

Browse Source
This commit is contained in:
Alinson S. Xavier
2025-07-22 12:25:47 -05:00
parent 596f41c477
commit bfaae7c005
1 changed files with 10 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
miplearn/collectors/basic.py
View File

@@ -17,9 +17,15 @@ from ..parallel import p_umap
class BasicCollector: class BasicCollector:
def __init__(self, skip_lp: bool = False, write_mps: bool = True) -> None: def __init__(
self,
skip_lp: bool = False,
write_mps: bool = True,
write_log: bool = True,
) -> None:
self.skip_lp = skip_lp self.skip_lp = skip_lp
self.write_mps = write_mps self.write_mps = write_mps
self.write_log = write_log
def collect( def collect(
self, self,
@@ -80,6 +86,7 @@ class BasicCollector:
log = streams[0].getvalue() log = streams[0].getvalue()
h5.put_scalar("mip_log", log) h5.put_scalar("mip_log", log)
if self.write_log:
with open(log_filename, "w") as log_file: with open(log_filename, "w") as log_file:
log_file.write(log) log_file.write(log)
except: except: