fix: enhance environment detection for Google Colab in Environment class

xability · Dec 18, 2024 · f8950f4 · f8950f4
1 parent f4a12c7
commit f8950f4
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/maidr/util/environment.py b/maidr/util/environment.py
@@ -1,5 +1,6 @@
 import json
 import os
+import sys
 
 
 class Environment:
@@ -24,7 +25,12 @@ def is_notebook() -> bool:
 
             return get_ipython() is not None and "ipykernel" in str(get_ipython())
         except ImportError:
-            return False
+            pass
+
+        if "google.colab" in sys.modules:
+            return True
+
+        return False
 
     @staticmethod
     def get_renderer() -> str: