change method of env check

pyproject.toml

@@ -45,7 +45,6 @@ python = ">= 3.8, <3.12"
 
 fsspec = ">= 2023.4.0"
 importlib-resources = { version = ">= 1.3", python = "<3.9" }
-ipython = ">= 8.21.0"
 jsonschema = ">= 4.18"
 pyarrow = ">= 11.0.0"
 pyyaml = ">= 5.3.1"

src/fondant/pipeline/lightweight_component.py

@@ -9,7 +9,6 @@
 from importlib import metadata
 
 import pyarrow as pa
-from IPython import get_ipython
 from IPython.core.magics.code import extract_symbols
 
 from fondant.component import BaseComponent, Component
@@ -287,9 +286,15 @@ def new_getfile(_object, _old_getfile=inspect.getfile):
     raise TypeError(msg)
 
 
-def is_running_in_jupyter():
-    shell = get_ipython().__class__.__name__
-    return shell == "ZMQInteractiveShell"
+def is_running_in_notebook():
+    """Check if the code is running in a Jupyter notebook."""
+    try:
+        from IPython import get_ipython
+
+        shell = get_ipython().__class__.__name__
+        return shell == "ZMQInteractiveShell"
+    except ModuleNotFoundError:
+        return False
 
 
 def build_python_script(component_cls: t.Type[Component]) -> str:
@@ -309,7 +314,7 @@ def build_python_script(component_cls: t.Type[Component]) -> str:
     """,
     )
 
-    if is_running_in_jupyter():
+    if is_running_in_notebook():
         inspect.getfile = new_getfile
         component_source = "".join(
             inspect.linecache.getlines(  # type: ignore[attr-defined]