[Bug] Fix for environment detection in barebones environments. (guida…

…nce-ai#1076)
svrvt · Nov 14, 2024 · 40a64d6 · 40a64d6
1 parent c6db4ac
commit 40a64d6
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