Create util for display jupyter version warnings and update imports

Author: marthacryan

packages/python/plotly/plotly/basewidget.py

@@ -1,6 +1,6 @@
 import pathlib
-import ipywidgets as widgets
-from traitlets import List, Unicode, Dict, observe, Integer
+from traitlets import List, Dict, observe, Integer
+from plotly.io._utils import display_jupyter_version_warnings
 
 from .basedatatypes import BaseFigure, BasePlotlyType
 from .callbacks import BoxSelector, LassoSelector, InputDeviceState, Points
@@ -725,6 +725,7 @@ def _repr_mimebundle_(self, include=None, exclude=None, validate=True, **kwargs)
         """
         Return mimebundle corresponding to default renderer.
         """
+        display_jupyter_version_warnings()
         return {
             "application/vnd.jupyter.widget-view+json": {
                 "version_major": 2,

packages/python/plotly/plotly/io/_renderers.py

@@ -2,8 +2,6 @@
 from copy import copy
 
 import os
-import warnings
-import psutil
 from packaging.version import Version
 
 from plotly import optional_imports
@@ -28,13 +26,14 @@
     CoCalcRenderer,
     DatabricksRenderer,
 )
-from plotly.io._utils import validate_coerce_fig_to_dict
+from plotly.io._utils import (
+    validate_coerce_fig_to_dict,
+    display_jupyter_version_warnings,
+)
 
 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
 # -----------------------------
@@ -398,18 +397,7 @@ 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`."
-            )
+        display_jupyter_version_warnings()
 
         ipython_display.display(bundle, raw=True)
 

packages/python/plotly/plotly/io/_utils.py

@@ -1,6 +1,29 @@
 import plotly
 import plotly.graph_objs as go
 from plotly.offline import get_plotlyjs_version
+from plotly import optional_imports
+
+import warnings
+import psutil
+
+jupyter_notebook = optional_imports.get_module("notebook")
+jupyter_lab = optional_imports.get_module("jupyterlab")
+
+
+def display_jupyter_version_warnings():
+
+    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`."
+        )
 
 
 def validate_coerce_fig_to_dict(fig, validate):

packages/python/plotly/test_requirements/requirements_312_np2_optional.txt

@@ -18,4 +18,5 @@ matplotlib==3.9.2
 scikit-image==0.24.0
 psutil==5.9.7
 kaleido
-orjson==3.9.10
+orjson==3.9.10
+anywidget==0.9.13