aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorJ08nY2025-03-22 10:43:45 +0100
committerJ08nY2025-04-16 12:25:06 +0200
commitc084f85cddde9727e61a503fd37af2551b94eeb8 (patch)
tree22060a07ffec759c4a6d10d88551e5e4fb3683fd
parenta7c2c44e6287addae82875ee23718b9466e5f703 (diff)
downloadECTester-c084f85cddde9727e61a503fd37af2551b94eeb8.tar.gz
ECTester-c084f85cddde9727e61a503fd37af2551b94eeb8.tar.zst
ECTester-c084f85cddde9727e61a503fd37af2551b94eeb8.zip
Fix probmaps script.
-rw-r--r--epare/probmaps.py5
1 files changed, 1 insertions, 4 deletions
diff --git a/epare/probmaps.py b/epare/probmaps.py
index c8431ff..980f9ed 100644
--- a/epare/probmaps.py
+++ b/epare/probmaps.py
@@ -93,7 +93,7 @@ if __name__ == "__main__":
for fname in files:
pool.submit_task(fname,
load_chunk,
- fname, selected_divisors, kind, use_init, use_multiply)
+ fname, selected_divisors, kind)
for fname, future in tqdm(pool.as_completed(), total=len(pool.tasks), smoothing=0):
if error := future.exception():
print(f"Error {fname}, {error}")
@@ -104,9 +104,6 @@ if __name__ == "__main__":
distributions_mults[mult].merge(prob_map)
else:
distributions_mults[mult] = prob_map
- # Save intermediate.
- with open(f"{divisor_name}_intermediate_{kind}_distrs.pickle", "wb") as f:
- pickle.dump(distributions_mults, f)
for mult, prob_map in distributions_mults.items():
print(f"Got {prob_map.samples} for {mult}.")
# Save