Skip to content

Commit c22bce4

Browse files
committed
Fix merge probs zstd error.
1 parent 942c110 commit c22bce4

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

analysis/scalarmults/epare/standalone/merge_probs.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -59,9 +59,9 @@ def main(compressed):
5959
maps[full].merge(prob_map)
6060
except EOFError:
6161
break
62-
except pickle.UnpicklingError:
62+
except (pickle.UnpicklingError, zstd.ZstdError):
6363
click.echo(
64-
f"Bad unpickling, the probs file {file} is likely truncated."
64+
f"Bad unpickling/decompression, the probs file {file} is likely truncated."
6565
)
6666
break
6767
click.echo(f"Loaded {i} probmaps from {file}.")

0 commit comments

Comments
 (0)