Fix for environment detection in barebones environments.

Missing import error catch in environment detection.
guidance-ai · Nov 11, 2024 · 2c47707 · 2c47707
1 parent c7e2fbc
commit 2c47707
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/guidance/visual/_environment.py b/guidance/visual/_environment.py
@@ -104,6 +104,8 @@ def _detect_ipython(flags: EnvFlags) -> bool:
         found = get_ipython() is not None
     except NameError:  # pragma: no cover
         pass
+    except ImportError:  # pragma: no cover
+        pass
     return found
 
 
@@ -131,6 +133,8 @@ def _detect_ipython_zmq(flags: EnvFlags) -> bool:
             found = False  # Other type (?)
     except NameError:  # pragma: no cover
         found = False  # Probably standard Python interpreter
+    except ImportError:  # pragma: no cover
+        found = False  # Probably standard Python interpreter
 
     flags.is_notebook |= found
     return found