From 4b1da729bc279be632a8dd9476eb80ec1c848381 Mon Sep 17 00:00:00 2001
From: Youngjoon Lee <5462944+youngjoon-lee@users.noreply.github.com>
Date: Wed, 7 Aug 2024 08:27:31 +0900
Subject: [PATCH] continue even if iteration failed, after saving err file

---
 mixnet/queuesim/queuesim.py | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/mixnet/queuesim/queuesim.py b/mixnet/queuesim/queuesim.py
index b28ca63..6d00fba 100644
--- a/mixnet/queuesim/queuesim.py
+++ b/mixnet/queuesim/queuesim.py
@@ -125,9 +125,18 @@ def run_session(
             try:
                 dissemination_times = future.result()
             except BaseException as e:
-                print(f"Error occurred in ParamSet-{paramset_id}, Iter-{iter_idx}: {e}")
+                msg = (
+                    f"Error occurred in ParamSet-{paramset_id}, IterIdx-{iter_idx}: {e}"
+                )
+                print(msg)
                 traceback.print_exc()
-                raise
+                with open(
+                    f"{outdir}/{subdir}/paramset_{paramset_id}_iteridx_{iter_idx}.err",
+                    "a",
+                ) as f:
+                    f.write(f"{msg}\n\n")
+                    traceback.print_exc(file=f)
+                continue  # skip the iteration failed
 
             paramset_results[paramset_id][0].add(iter_idx)
             paramset_results[paramset_id][1].extend(dissemination_times)