fix traceback handling

Signed-off-by: Panos Vagenas <35837085+vagenas@users.noreply.github.com>
DS4SD · Jun 10, 2024 · 07635e3 · 07635e3
1 parent dcbbf00
commit 07635e3
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/nbrunner/nb_runner.py b/nbrunner/nb_runner.py
@@ -1,5 +1,6 @@
 import glob
 import os
+import traceback
 import uuid
 from pathlib import Path
 from typing import Optional, Set
@@ -56,9 +57,9 @@ def execute_notebook(self, run_id, notebook_path):
         )
         try:
             out = ep.preprocess(nb, {"metadata": {"path": notebook_path.parent}})
-        except (CellExecutionError, CellTimeoutError) as e:
+        except (CellExecutionError, CellTimeoutError):
             print(f"=> Error during {run_id}; check {output_filename} for details")
-            print(e.traceback)
+            print(traceback.print_exc())
             out = None
         finally:
             with open(output_filename, mode="w", encoding="utf-8") as f: