Improve notebook detection (#5792)

holoviz · Jul 5, 2023 · 53111a9 · 53111a9
1 parent 3c751b8
commit 53111a9
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/holoviews/util/__init__.py b/holoviews/util/__init__.py
@@ -713,14 +713,11 @@ def __call__(self, *args, **params):
         import panel as pn
 
         if pn.config.comms == "default":
-            try:
-                import google.colab  # noqa
+            if "google.colab" in sys.modules:
                 pn.config.comms = "colab"
                 return
-            except ImportError:
-                pass
 
-            if "VSCODE_PID" in os.environ:
+            if "VSCODE_CWD" in os.environ or "VSCODE_PID" in os.environ:
                 pn.config.comms = "vscode"
                 self._ignore_bokeh_warnings()
                 return