Merge pull request #5573 from plotly/ekl-migrate-ci-uv-lock-upgrade

Upgrade dependency versions in `uv.lock`, test with Pandas 3 in CI

Author: emilykl

.github/workflows/check-js-build.yml

@@ -1,4 +1,13 @@
-on: push
+on:
+  push:
+    branches:
+      - main
+  pull_request:
+    types: [opened, reopened]
+    paths:
+      - 'js/**'
+      - 'plotly/labextension/**'
+
 name: Check JS build
 
 jobs:

.github/workflows/run-pytest.yml

@@ -56,7 +56,7 @@ jobs:
         run: |
           uv venv
           source .venv/bin/activate
-          uv sync --extra dev_optional
+          uv sync --extra dev_optional --extra dev_pandas3
           python --version
       - name: Test core
         run: |
@@ -84,14 +84,22 @@ jobs:
           python -m pytest -x test_init/test_lazy_imports.py
 
   test-optional-legacy-pandas:
-    name: Optional tests, Pandas 1 (Python 3.9, Pandas 1.2.4)
+    name: Optional tests (Python ${{ matrix.python-version }}, pandas ${{ matrix.pandas-version }})
     runs-on: ubuntu-latest
+    strategy:
+      fail-fast: false
+      matrix:
+        include:
+          - python-version: "3.9"
+            pandas-version: "1"
+          - python-version: "3.11"
+            pandas-version: "2"
     steps:
       - uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2
       - name: Set up Python
         uses: actions/setup-python@a309ff8b426b58ec0e2a45f0f869d46889d02405  # v6.2.0
         with:
-          python-version: "3.9"
+          python-version: ${{ matrix.python-version }}
       - name: Set up Chrome for browser tests
         uses: ./.github/actions/setup-chrome-for-pytest
       - name: Set up uv
@@ -100,8 +108,7 @@ jobs:
         run: |
           uv venv
           source .venv/bin/activate
-          uv sync --extra dev_optional
-          uv pip install pandas==1.2.4 numpy==1.26.4
+          uv sync --extra dev_optional --extra dev_pandas${{ matrix.pandas-version }}
       - name: Test core
         run: |
           source .venv/bin/activate
@@ -128,7 +135,7 @@ jobs:
           python -m pytest -x test_init/test_lazy_imports.py
 
   test-kaleido-v0:
-    name: Optional tests (Kaleido only), Kaleido v0 (Python 3.12, Kaleido v0.2.1)
+    name: Optional tests (Kaleido only), Kaleido v0 (Python 3.12, kaleido 0.2.1)
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd  # v6.0.2

CHANGELOG.md

@@ -4,6 +4,9 @@ This project adheres to [Semantic Versioning](http://semver.org/).
 
 ## Unreleased
 
+### Fixed
+- Update tests to be compatible with numpy 2.4 [[#5522](https://github.com/plotly/plotly.py/pull/5522)], with thanks to @thunze for the contribution!
+
 
 ## [6.7.0] - 2026-04-09
 

pyproject.toml

@@ -83,8 +83,26 @@ dev = [
     "plotly[dev_optional]"
 ]
 
+# The following extras are intended for use in CI only
+dev_pandas1 = [
+    "pandas>=1,<2",
+    "numpy>=1,<2",
+    "setuptools<82"
+]
+dev_pandas2 = [
+    "pandas>=2,<3",
+]
+dev_pandas3 = [
+    "pandas>=3;python_version>='3.11'",
+]
+
 [tool.uv]
 exclude-newer = "72 hours"
+conflicts = [
+    [{ extra = "dev_pandas1" }, { extra = "dev_pandas2" }],
+    [{ extra = "dev_pandas1" }, { extra = "dev_pandas3" }],
+    [{ extra = "dev_pandas2" }, { extra = "dev_pandas3" }],
+]
 
 [project.scripts]
 plotly_get_chrome = "plotly.io._kaleido:plotly_get_chrome"

tests/test_optional/test_px/test_px_functions.py

@@ -1,11 +1,19 @@
-import plotly.express as px
-import plotly.graph_objects as go
-from numpy.testing import assert_array_equal
 import narwhals.stable.v1 as nw
 import numpy as np
+from numpy.testing import assert_array_equal
+from packaging.version import Version
+import pandas
+import plotly.express as px
+import plotly.graph_objects as go
 from polars.exceptions import InvalidOperationError
 import pytest
 
+from .conftest import pandas_pyarrow_constructor
+
+
+def _pandas_version_at_least(version: str) -> bool:
+    return Version(pandas.__version__) >= Version(version)
+
 
 def _compare_figures(go_trace, px_fig):
     """Compare a figure created with a go trace and a figure created with
@@ -153,6 +161,11 @@ def test_sunburst_treemap_colorscales():
 
 
 def test_sunburst_treemap_with_path(constructor):
+    if _pandas_version_at_least("3.0.0") and constructor == pandas_pyarrow_constructor:
+        pytest.skip(
+            "known issue with pandas 3 + pandas_pyarrow_constructor + px.sunburst() (https://github.com/plotly/plotly.py/issues/5571)"
+        )
+
     vendors = ["A", "B", "C", "D", "E", "F", "G", "H"]
     sectors = [
         "Tech",
@@ -249,6 +262,11 @@ def test_sunburst_treemap_with_path_and_hover(backend):
 
 
 def test_sunburst_treemap_with_path_color(constructor):
+    if _pandas_version_at_least("3.0.0") and constructor == pandas_pyarrow_constructor:
+        pytest.skip(
+            "known issue with pandas 3 + pandas_pyarrow_constructor + px.sunburst() (https://github.com/plotly/plotly.py/issues/5571)"
+        )
+
     vendors = ["A", "B", "C", "D", "E", "F", "G", "H"]
     sectors = [
         "Tech",
@@ -327,6 +345,11 @@ def test_sunburst_treemap_with_path_color(constructor):
 
 
 def test_sunburst_treemap_column_parent(constructor):
+    if _pandas_version_at_least("3.0.0") and constructor == pandas_pyarrow_constructor:
+        pytest.skip(
+            "known issue with pandas 3 + pandas_pyarrow_constructor + px.sunburst() (https://github.com/plotly/plotly.py/issues/5571)"
+        )
+
     vendors = ["A", "B", "C", "D", "E", "F", "G", "H"]
     sectors = [
         "Tech",
@@ -354,6 +377,11 @@ def test_sunburst_treemap_column_parent(constructor):
 
 
 def test_sunburst_treemap_with_path_non_rectangular(constructor):
+    if _pandas_version_at_least("3.0.0") and constructor == pandas_pyarrow_constructor:
+        pytest.skip(
+            "known issue with pandas 3 + pandas_pyarrow_constructor + px.sunburst() (https://github.com/plotly/plotly.py/issues/5571)"
+        )
+
     vendors = ["A", "B", "C", "D", None, "E", "F", "G", "H", None]
     sectors = [
         "Tech",

tests/test_optional/test_utils/test_utils.py

@@ -311,11 +311,24 @@ def test_encode_customdata_datetime_series(self):
         fig_json = _json.dumps(
             fig, cls=utils.PlotlyJSONEncoder, separators=(",", ":"), sort_keys=True
         )
-        self.assertTrue(
-            fig_json.startswith(
-                '{"data":[{"customdata":["2010-01-01T00:00:00.000000000","2010-01-02T00:00:00.000000000"]'
+
+        fig_from_json = Figure(_json.loads(fig_json))
+
+        import pandas
+
+        if Version(pandas.__version__) >= Version("3.0.0"):
+            # Starting in pandas 3, datetimes have ms precision by default
+            # https://pandas.pydata.org/docs/whatsnew/v3.0.0.html#datetime-timedelta-resolution-inference
+            assert fig_from_json.data[0].customdata == (
+                "2010-01-01T00:00:00.000000",
+                "2010-01-02T00:00:00.000000",
+            )
+        else:
+            # Before pandas 3, datetimes have ns precision by default
+            assert fig_from_json.data[0].customdata == (
+                "2010-01-01T00:00:00.000000000",
+                "2010-01-02T00:00:00.000000000",
             )
-        )
 
     def test_encode_customdata_datetime_homogeneous_dataframe(self):
         df = pd.DataFrame(
@@ -332,13 +345,24 @@ def test_encode_customdata_datetime_homogeneous_dataframe(self):
         fig_json = _json.dumps(
             fig, cls=utils.PlotlyJSONEncoder, separators=(",", ":"), sort_keys=True
         )
-        self.assertTrue(
-            fig_json.startswith(
-                '{"data":[{"customdata":'
-                '[["2010-01-01T00:00:00.000000000","2011-01-01T00:00:00.000000000"],'
-                '["2010-01-02T00:00:00.000000000","2011-01-02T00:00:00.000000000"]'
+
+        fig_from_json = Figure(_json.loads(fig_json))
+
+        import pandas
+
+        if Version(pandas.__version__) >= Version("3.0.0"):
+            # Starting in pandas 3, datetimes have ms precision by default
+            # https://pandas.pydata.org/docs/whatsnew/v3.0.0.html#datetime-timedelta-resolution-inference
+            assert fig_from_json.data[0].customdata == (
+                ["2010-01-01T00:00:00.000000", "2011-01-01T00:00:00.000000"],
+                ["2010-01-02T00:00:00.000000", "2011-01-02T00:00:00.000000"],
+            )
+        else:
+            # Before pandas 3, datetimes have ns precision by default
+            assert fig_from_json.data[0].customdata == (
+                ["2010-01-01T00:00:00.000000000", "2011-01-01T00:00:00.000000000"],
+                ["2010-01-02T00:00:00.000000000", "2011-01-02T00:00:00.000000000"],
             )
-        )
 
     def test_encode_customdata_datetime_inhomogeneous_dataframe(self):
         df = pd.DataFrame(
@@ -384,11 +408,26 @@ def test_datetime_dot_date(self):
     def test_numpy_datetime64(self):
         a = pd.date_range("2011-07-11", "2011-07-13", freq="D").values
         j1 = _json.dumps(a, cls=utils.PlotlyJSONEncoder)
-        assert (
-            j1 == '["2011-07-11T00:00:00.000000000", '
-            '"2011-07-12T00:00:00.000000000", '
-            '"2011-07-13T00:00:00.000000000"]'
-        )
+
+        from_json = _json.loads(j1)
+
+        import pandas
+
+        if Version(pandas.__version__) >= Version("3.0.0"):
+            # Starting in pandas 3, datetimes have ms precision by default
+            # https://pandas.pydata.org/docs/whatsnew/v3.0.0.html#datetime-timedelta-resolution-inference
+            assert from_json == [
+                "2011-07-11T00:00:00.000000",
+                "2011-07-12T00:00:00.000000",
+                "2011-07-13T00:00:00.000000",
+            ]
+        else:
+            # Before pandas 3, datetimes have ns precision by default
+            assert from_json == [
+                "2011-07-11T00:00:00.000000000",
+                "2011-07-12T00:00:00.000000000",
+                "2011-07-13T00:00:00.000000000",
+            ]
 
     def test_pil_image_encoding(self):
         img_path = os.path.join(