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

Make read/write_pickle_gz quiet

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-06-28 10:17:41 -05:00
parent 46a7d3fe26
commit 101bd94a5b
1 changed files with 2 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
miplearn/instance/picklegz.py
View File

@@ -134,17 +134,13 @@ class PickleGzInstance(Instance):
write_pickle_gz(self.instance, self.filename) write_pickle_gz(self.instance, self.filename)
def write_pickle_gz(obj: Any, filename: str, quiet: bool = False) -> None: def write_pickle_gz(obj: Any, filename: str) -> None:
if not quiet:
logger.info(f"Writing: {filename}")
os.makedirs(os.path.dirname(filename), exist_ok=True) os.makedirs(os.path.dirname(filename), exist_ok=True)
with gzip.GzipFile(filename, "wb") as file: with gzip.GzipFile(filename, "wb") as file:
pickle.dump(obj, cast(IO[bytes], file)) pickle.dump(obj, cast(IO[bytes], file))
def read_pickle_gz(filename: str, quiet: bool = False) -> Any: def read_pickle_gz(filename: str) -> Any:
if not quiet:
logger.info(f"Reading: {filename}")
with gzip.GzipFile(filename, "rb") as file: with gzip.GzipFile(filename, "rb") as file:
return pickle.load(cast(IO[bytes], file)) return pickle.load(cast(IO[bytes], file))