Make read/write_pickle_gz quiet

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

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

Loading…
Cancel
Save