feat: Allow exceptions in experiments to cause breakpoints in debug mode

Jacob-Stevens-Haas · Feb 29, 2024 · 5932a29 · 5932a29
1 parent 8995c6e
commit 5932a29
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/mitosis/__init__.py b/mitosis/__init__.py
@@ -496,10 +496,14 @@ def _run_in_notebook(
     ep = ExecutePreprocessor(timeout=-1, kernel=kernel_name)
     exception = None
     metrics = None
+    if debug:
+        allowed = ()
+    else:
+        allowed = (nbclient.exceptions.CellExecutionError,)
     try:
         ep.preprocess(nb, {"metadata": {"path": trials_folder}})
         metrics = nb["cells"][-1]["outputs"][0]["text"][:-1]
-    except nbclient.exceptions.CellExecutionError as exc:
+    except allowed as exc:
         exception = exc
     return nb, metrics, exception