From 989c971d8e6876b206d3f6223e51bad41b79620b Mon Sep 17 00:00:00 2001
From: jessegrabowski <jessegrabowski@gmail.com>
Date: Wed, 29 Jul 2026 23:05:39 -0500
Subject: [PATCH] bump pymc/pytensor pins

---
 conda_envs/environment-docs.yaml | 4 ++--
 conda_envs/environment-test.yaml | 4 ++--
 conda_envs/environment.yaml      | 4 ++--
 pyproject.toml                   | 4 ++--
 tests/test_vff_paths.py          | 4 ++--
 5 files changed, 10 insertions(+), 10 deletions(-)

diff --git a/conda_envs/environment-docs.yaml b/conda_envs/environment-docs.yaml
index b813c99..4434d5c 100644
--- a/conda_envs/environment-docs.yaml
+++ b/conda_envs/environment-docs.yaml
@@ -12,8 +12,8 @@ dependencies:
     - scipy
     - libblas=*=*mkl
     - mkl-service
-    - pymc>=6.1.0,<6.2
-    - pytensor>=3.1.2,<3.2
+    - pymc>=6.2.0,<6.3
+    - pytensor>=3.2.3,<3.3
     - matplotlib
     # Docs build deps.
     - ipython
diff --git a/conda_envs/environment-test.yaml b/conda_envs/environment-test.yaml
index ba6965e..081e5f9 100644
--- a/conda_envs/environment-test.yaml
+++ b/conda_envs/environment-test.yaml
@@ -4,8 +4,8 @@ channels:
 dependencies:
     - python>=3.12
     - numpy
-    - pymc>=6.1.0,<6.2
-    - pytensor>=3.1.2,<3.2
+    - pymc>=6.2.0,<6.3
+    - pytensor>=3.2.3,<3.3
     - pytest
     - mypy
     - polars
diff --git a/conda_envs/environment.yaml b/conda_envs/environment.yaml
index 83b3bff..088239f 100644
--- a/conda_envs/environment.yaml
+++ b/conda_envs/environment.yaml
@@ -3,8 +3,8 @@ channels:
     - conda-forge
 dependencies:
     - python>=3.12
-    - pymc>=6.1.0,<6.2
-    - pytensor>=3.1.2,<3.2
+    - pymc>=6.2.0,<6.3
+    - pytensor>=3.2.3,<3.3
     - jax
     - jaxlib
     - optax
diff --git a/pyproject.toml b/pyproject.toml
index a4d73a1..ffe2e2d 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -23,8 +23,8 @@ license = {file = "LICENSE"}
 dynamic = ["version"]
 dependencies = [
   "numpy",
-  "pymc>=6.1.0,<6.2",
-  "pytensor>=3.1.2,<3.2"
+  "pymc>=6.2.0,<6.3",
+  "pytensor>=3.2.3,<3.3"
 ]
 
 [project.optional-dependencies]
diff --git a/tests/test_vff_paths.py b/tests/test_vff_paths.py
index c67fcfd..4a4b8b3 100644
--- a/tests/test_vff_paths.py
+++ b/tests/test_vff_paths.py
@@ -103,10 +103,10 @@ def test_no_dense_cholesky_on_Kuu():
     fmean, fvar = svgp.predict_marginal(X_sym)
     fn = pytensor.function([X_sym], [fmean, fvar])
     fg = fn.maker.fgraph
-    shape_of = fg.shape_feature.shape_of
+    shape_feature = fg.shape_feature
     for node in fg.toposort():
         if node.op.__class__.__name__ == "Cholesky":
-            in_shape = shape_of[node.inputs[0]]
+            in_shape = shape_feature.shape_tuple(node.inputs[0])
             resolved = tuple(int(s.data) if hasattr(s, "data") else None for s in in_shape)
             assert resolved != (M, M), f"dense MxM Cholesky on Kuu at {node}"