Handle release of pymc 5.25.0 (#1844)

Author: daniel-saunders-phil

.github/workflows/pypi.yml

@@ -17,7 +17,7 @@ jobs:
           fetch-depth: 0
       - uses: actions/setup-python@v5
         with:
-          python-version: "3.10"
+          python-version: "3.11"
       - name: Build the sdist and the wheel
         run: |
           pip install build
@@ -62,7 +62,7 @@ jobs:
           repository-url: https://test.pypi.org/legacy/
       - uses: actions/setup-python@v5
         with:
-          python-version: "3.10"
+          python-version: "3.11"
       - name: Test pip install from test.pypi
         run: |
           python -m venv venv-test-pypi

.github/workflows/test.yml

@@ -30,7 +30,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        config: [ {python-version: "3.10", oldest-pymc: false}, {python-version: "3.12", oldest-pymc: true}]
+        config: [ {python-version: "3.11", oldest-pymc: false}, {python-version: "3.13", oldest-pymc: true}]
         split:
           - "--ignore tests/mmm --ignore tests/clv"
           - "tests/mmm --ignore tests/mmm/test_tvp.py --ignore tests/mmm/test_budget_optimizer.py --ignore tests/mmm/test_hsgp.py --ignore tests/mmm/test_transformers.py"
@@ -82,7 +82,7 @@ jobs:
       - name: Set up Python
         uses: actions/setup-python@v5
         with:
-          python-version: "3.10"
+          python-version: "3.11"
           cache: "pip"
           cache-dependency-path: "pyproject.toml"
       - name: Run tests

.github/workflows/uml.yml

@@ -18,7 +18,7 @@ jobs:
       - name: Set up Python
         uses: actions/setup-python@v5
         with:
-          python-version: "3.10"
+          python-version: "3.13"
 
       - name: Configure Git Identity
         run: |

.readthedocs.yml

@@ -6,9 +6,9 @@ sphinx:
   configuration: docs/source/conf.py
 
 build:
-  os: ubuntu-22.04
+  os: ubuntu-24.04
   tools:
-    python: "3.10"
+    python: "3.11"
   jobs:
     pre_build:
       # Install core dependencies first

environment.yml

@@ -13,7 +13,7 @@ dependencies:
 - streamlit>=1.25.0
 - pip
 - pydantic
-- preliz
+- preliz>=0.20.0
 - pyprojroot
 # NOTE: Keep minimum pymc version in sync with ci.yml `OLDEST_PYMC_VERSION`
 - pymc>=5.23.0

pymc_marketing/mmm/budget_optimizer.py

@@ -25,7 +25,6 @@
 from arviz import InferenceData
 from pydantic import BaseModel, ConfigDict, Field, InstanceOf
 from pymc import Model, do
-from pymc.logprob.utils import rvs_in_graph
 from pymc.model.transform.optimization import freeze_dims_and_data
 from pymc.pytensorf import rewrite_pregrad
 from pytensor import clone_replace, function
@@ -35,6 +34,11 @@
 from scipy.optimize import OptimizeResult, minimize
 from xarray import DataArray
 
+try:
+    from pymc.pytensorf import rvs_in_graph
+except ImportError:
+    from pymc.logprob.utils import rvs_in_graph
+
 from pymc_marketing.mmm.constraints import (
     Constraint,
     build_default_sum_constraint,

pyproject.toml

@@ -13,9 +13,9 @@ classifiers = [
     "Development Status :: 5 - Production/Stable",
     "Programming Language :: Python",
     "Programming Language :: Python :: 3",
-    "Programming Language :: Python :: 3.10",
     "Programming Language :: Python :: 3.11",
     "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: 3.13",
     "License :: OSI Approved :: Apache Software License",
     "Intended Audience :: Science/Research",
     "Topic :: Scientific/Engineering",
@@ -38,7 +38,7 @@ dependencies = [
     "xarray>=2024.1.0",
     "xarray-einstats>=0.5.1",
     "pyprojroot",
-    "preliz>=0.8.0",
+    "preliz>=0.20.0",
 ]
 
 [project.optional-dependencies]
@@ -76,6 +76,7 @@ test = [
     "blackjax",
     "nutpie",
     "numpyro",
+    "jax<0.7.0", # related to https://github.com/pyro-ppl/numpyro/issues/2051
     "graphviz>=0.20.1",
     "lifetimes==0.11.3",
     "papermill",