Add warning message when notebook version < 7 is used

marthacryan · marthacryan · commit e7b6d7ab4025 · 2024-10-11T15:13:21.000-05:00
diff --git a/packages/python/plotly/plotly/io/_renderers.py b/packages/python/plotly/plotly/io/_renderers.py
@@ -2,6 +2,8 @@
 from copy import copy
 
 import os
+import warnings
+import psutil
 from packaging.version import Version
 
 from plotly import optional_imports
@@ -31,7 +33,8 @@
 ipython = optional_imports.get_module("IPython")
 ipython_display = optional_imports.get_module("IPython.display")
 nbformat = optional_imports.get_module("nbformat")
-
+jupyter_notebook = optional_imports.get_module("notebook")
+jupyter_lab = optional_imports.get_module("jupyterlab")
 
 # Renderer configuration class
 # -----------------------------
@@ -395,6 +398,19 @@ def show(fig, renderer=None, validate=True, **kwargs):
                 "Mime type rendering requires nbformat>=4.2.0 but it is not installed"
             )
 
+        parent_process = psutil.Process().parent().cmdline()[-1]
+
+        if "jupyter-notebook" in parent_process and jupyter_notebook.__version__ < "7":
+            # Add warning about upgrading notebook
+            warnings.warn(
+                f"Plotly version >= 6 requires Jupyter Notebook >= 7 but you have {jupyter_notebook.__version__} installed.\n To upgrade Jupyter Notebook, please run `pip install notebook --upgrade`."
+            )
+        elif "jupyter-lab" in parent_process and jupyter_lab.__version__ < "3":
+            # Add warning about upgrading jupyterlab
+            warnings.warn(
+                f"Plotly version >= 6 requires JupyterLab >= 3 but you have {jupyter_lab.__version__} installed. To upgrade JupyterLab, please run `pip install jupyterlab --upgrade`."
+            )
+
         ipython_display.display(bundle, raw=True)
 
     # external renderers
