From bfaae7c00506b3104c581ddb5e2f50b61c229899 Mon Sep 17 00:00:00 2001 From: "Alinson S. Xavier" Date: Tue, 22 Jul 2025 12:25:47 -0500 Subject: [PATCH] BasicCollector: Make log file optional --- miplearn/collectors/basic.py | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/miplearn/collectors/basic.py b/miplearn/collectors/basic.py index ea40b78..beb76f5 100644 --- a/miplearn/collectors/basic.py +++ b/miplearn/collectors/basic.py @@ -17,9 +17,15 @@ from ..parallel import p_umap 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.write_mps = write_mps + self.write_log = write_log def collect( self, @@ -80,8 +86,9 @@ class BasicCollector: log = streams[0].getvalue() h5.put_scalar("mip_log", log) - with open(log_filename, "w") as log_file: - log_file.write(log) + if self.write_log: + with open(log_filename, "w") as log_file: + log_file.write(log) except: print(f"Error processing: data_filename") traceback.print_exc()