fix: is_notebook returns false for google colab

xability · Dec 18, 2024 · a2f1148 · a2f1148
1 parent 229a3ac
commit a2f1148
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/maidr/util/environment.py b/maidr/util/environment.py
@@ -22,7 +22,10 @@ def is_notebook() -> bool:
         try:
             from IPython import get_ipython  # type: ignore
 
-            return get_ipython() is not None and "ipykernel" in str(get_ipython())
+            return get_ipython() is not None and (
+                "ipykernel" in str(get_ipython())
+                or "google.colab" in str(get_ipython())
+            )
         except ImportError:
             return False