From 101bd94a5bf66545fcdcd1a2a80888ed0468ccd8 Mon Sep 17 00:00:00 2001 From: "Alinson S. Xavier" Date: Mon, 28 Jun 2021 10:17:41 -0500 Subject: [PATCH] Make read/write_pickle_gz quiet --- miplearn/instance/picklegz.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/miplearn/instance/picklegz.py b/miplearn/instance/picklegz.py index 83495e1..720f390 100644 --- a/miplearn/instance/picklegz.py +++ b/miplearn/instance/picklegz.py @@ -134,17 +134,13 @@ class PickleGzInstance(Instance): write_pickle_gz(self.instance, self.filename) -def write_pickle_gz(obj: Any, filename: str, quiet: bool = False) -> None: - if not quiet: - logger.info(f"Writing: {filename}") +def write_pickle_gz(obj: Any, filename: str) -> None: os.makedirs(os.path.dirname(filename), exist_ok=True) with gzip.GzipFile(filename, "wb") as file: pickle.dump(obj, cast(IO[bytes], file)) -def read_pickle_gz(filename: str, quiet: bool = False) -> Any: - if not quiet: - logger.info(f"Reading: {filename}") +def read_pickle_gz(filename: str) -> Any: with gzip.GzipFile(filename, "rb") as file: return pickle.load(cast(IO[bytes], file))