[Bug] Fix for environment detection in barebones environments. (#1076)

guidance-ai · Nov 11, 2024 · 13ef66e · 13ef66e
1 parent c7e2fbc
commit 13ef66e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/guidance/visual/_environment.py b/guidance/visual/_environment.py
@@ -102,7 +102,7 @@ def _detect_ipython(flags: EnvFlags) -> bool:
     try:
         from IPython import get_ipython
         found = get_ipython() is not None
-    except NameError:  # pragma: no cover
+    except (NameError, ImportError):  # pragma: no cover
         pass
     return found
 
@@ -129,7 +129,7 @@ def _detect_ipython_zmq(flags: EnvFlags) -> bool:
             found = False  # Terminal running IPython
         else:
             found = False  # Other type (?)
-    except NameError:  # pragma: no cover
+    except (NameError, ImportError):  # pragma: no cover
         found = False  # Probably standard Python interpreter
 
     flags.is_notebook |= found