1 files changed, 12 insertions, 2 deletions

diff --git a/epare/probmaps.py b/epare/probmaps.py
index cf1a050..a5d4510 100644
--- a/epare/probmaps.py
+++ b/epare/probmaps.py
@@ -15,6 +15,7 @@ def divides_any(l: int, small_scalars: set[int]) -> bool:
             return True
     return False
 
+
 def process_small_scalars(scalar_results: MultResults, divisors: set[int]) -> ProbMap:
     result = {}
     for divisor in divisors:
@@ -25,18 +26,27 @@ def process_small_scalars(scalar_results: MultResults, divisors: set[int]) -> Pr
         result[divisor] = count / scalar_results.samples
     return ProbMap(result, scalar_results.samples, scalar_results.kind)
 
+
 def load_chunk(fname: str, divisors: set[int], kind: str) -> dict[MultIdent, ProbMap]:
     with open(fname, "rb") as f:
-        multiples = pickle.load(f)
+        multiples = {}
+        while True:
+            try:
+                mult, distr = pickle.load(f)
+                multiples[mult] = distr
+            except EOFError:
+                break
         res = {}
         for mult, results in multiples.items():
             results.kind = kind
             res[mult] = process_small_scalars(results, divisors)
     return res
 
+
 def powers_of(k, max_power=20):
     return [k**i for i in range(1, max_power)]
 
+
 def prod_combine(one, other):
     return [a * b for a, b in itertools.product(one, other)]
 
@@ -99,4 +109,4 @@ if __name__ == "__main__":
         print(f"Got {prob_map.samples} for {mult}.")
     # Save
     with open(f"{divisor_name}_{kind}_distrs.pickle", "wb") as f:
-        pickle.dump(distributions_mults, f)
\ No newline at end of file
+        pickle.dump(distributions_mults, f)