Skip to content

Commit 4d33fd6

Browse files
committed
force gzip for pkls
1 parent ba51063 commit 4d33fd6

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

pypop/prv.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -238,7 +238,7 @@ def save(self, filename):
238238
pickle.dump(savedata, fh)
239239

240240
def _load_pickle(self, filename):
241-
with zipopen(filename, "rb") as fh:
241+
with gzip.open(filename, "rb") as fh:
242242
data = pickle.load(fh)
243243

244244
try:
@@ -262,7 +262,7 @@ def profile_openmp_regions(self, no_progress=False, ignore_cache=False):
262262

263263
if not ignore_cache:
264264
try:
265-
with zipopen(
265+
with gzip.open(
266266
PRV._generate_region_cache_name(self._prv_path), "rb"
267267
) as fh:
268268
self._omp_region_data = pickle.load(fh)
@@ -466,7 +466,7 @@ def profile_openmp_regions(self, no_progress=False, ignore_cache=False):
466466

467467
self._omp_region_data = pd.concat(rank_stats, names=["rank", "region"])
468468

469-
with zipopen(PRV._generate_region_cache_name(self._prv_path), "wb") as fh:
469+
with gzip.open(PRV._generate_region_cache_name(self._prv_path), "wb") as fh:
470470
pickle.dump(self._omp_region_data, fh)
471471

472472
return self._omp_region_data

0 commit comments

Comments
 (0)