raise exception instead (#1862)

Netflix · May 27, 2024 · b512656 · b512656
1 parent 34ecc5a
commit b512656
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/metaflow/runner/nbrun.py b/metaflow/runner/nbrun.py
@@ -8,6 +8,12 @@
 DEFAULT_DIR = tempfile.gettempdir()
 
 
+class NBRunnerInitializationError(Exception):
+    """Custom exception for errors during NBRunner initialization."""
+
+    pass
+
+
 def get_current_cell(ipython):
     if ipython:
         return ipython.history_manager.input_hist_raw[-1]
@@ -53,10 +59,9 @@ def __init__(
 
             ipython = get_ipython()
         except ModuleNotFoundError:
-            print(
-                "'nbrun' requires an interactive python environment (such as Jupyter)"
+            raise NBRunnerInitializationError(
+                "'NBRunner' requires an interactive Python environment (such as Jupyter)"
             )
-            return
 
         self.cell = get_current_cell(ipython)
         self.flow = flow