Bump minor PyTensor dependency

Author: ricardoV94

conda-envs/environment-alternative-backends.yml

@@ -22,7 +22,7 @@ dependencies:
 - numpyro>=0.8.0
 - pandas>=0.24.0
 - pip
-- pytensor>=2.36.0,<2.37
+- pytensor>=2.36.1,<2.37
 - python-graphviz
 - networkx
 - rich>=13.7.1

conda-envs/environment-dev.yml

@@ -12,7 +12,7 @@ dependencies:
 - numpy>=1.25.0
 - pandas>=0.24.0
 - pip
-- pytensor>=2.36.0,<2.37
+- pytensor>=2.36.1,<2.37
 - python-graphviz
 - networkx
 - scipy>=1.4.1

conda-envs/environment-docs.yml

@@ -11,7 +11,7 @@ dependencies:
 - numpy>=1.25.0
 - pandas>=0.24.0
 - pip
-- pytensor>=2.36.0,<2.37
+- pytensor>=2.36.1,<2.37
 - python-graphviz
 - rich>=13.7.1
 - scipy>=1.4.1

conda-envs/environment-test.yml

@@ -14,7 +14,7 @@ dependencies:
 - pandas>=0.24.0
 - pip
 - polyagamma
-- pytensor>=2.36.0,<2.37
+- pytensor>=2.36.1,<2.37
 - python-graphviz
 - networkx
 - rich>=13.7.1

conda-envs/windows-environment-dev.yml

@@ -12,7 +12,7 @@ dependencies:
 - numpy>=1.25.0
 - pandas>=0.24.0
 - pip
-- pytensor>=2.36.0,<2.37
+- pytensor>=2.36.1,<2.37
 - python-graphviz
 - networkx
 - rich>=13.7.1

conda-envs/windows-environment-test.yml

@@ -15,7 +15,7 @@ dependencies:
 - pandas>=0.24.0
 - pip
 - polyagamma
-- pytensor>=2.36.0,<2.37
+- pytensor>=2.36.1,<2.37
 - python-graphviz
 - networkx
 - rich>=13.7.1

pymc/distributions/multivariate.py

@@ -2264,11 +2264,11 @@ class CAR(Continuous):
     def dist(cls, mu, W, alpha, tau, *args, **kwargs):
         # This variable has an expensive validation check, that we want to constant-fold if possible
         # So it's passed as an explicit input
-        from pytensor.sparse import as_sparse_or_tensor_variable, structured_sign
+        from pytensor.sparse import as_sparse_or_tensor_variable, sign
 
         W = as_sparse_or_tensor_variable(W)
         if isinstance(W.type, pytensor.sparse.SparseTensorType):
-            abs_diff = structured_sign(W - W.T) * (W - W.T)
+            abs_diff = sign(W - W.T) * (W - W.T)
             W_is_valid = pt.isclose(abs_diff.sum(), 0)
         else:
             W_is_valid = pt.allclose(W, W.T)

requirements-dev.txt

@@ -16,7 +16,7 @@ pandas>=0.24.0
 polyagamma
 pre-commit>=2.8.0
 pymc-sphinx-theme>=0.16.0
-pytensor>=2.36.0,<2.37
+pytensor>=2.36.1,<2.37
 pytest-cov>=2.5
 pytest>=3.0
 rich>=13.7.1

requirements.txt

@@ -3,7 +3,7 @@ cachetools>=4.2.1
 cloudpickle
 numpy>=1.25.0
 pandas>=0.24.0
-pytensor>=2.36.0,<2.37
+pytensor>=2.36.1,<2.37
 rich>=13.7.1
 scipy>=1.4.1
 threadpoolctl>=3.1.0,<4.0.0

tests/distributions/test_custom.py

@@ -356,24 +356,25 @@ def test_custom_dist_default_support_point(self, dist_params, size, expected, di
         assert_support_point_is_expected(model, expected)
 
     def test_custom_dist_default_support_point_scan(self):
-        def scan_step(left, right):
-            x = Uniform.dist(left, right)
-            x_update = collect_default_updates([x])
-            return x, x_update
+        def scan_step(left, right, rng):
+            x = Uniform.dist(left, right, rng=rng)
+            x_update = collect_default_updates([x], must_be_shared=False)
+            return x, x_update[rng]
 
         def dist(size):
-            with pytest.warns(DeprecationWarning, match="Scan return signature will change"):
-                xs, updates = scan(
-                    fn=scan_step,
-                    sequences=[
-                        pt.as_tensor_variable(np.array([-4, -3])),
-                        pt.as_tensor_variable(np.array([-2, -1])),
-                    ],
-                    name="xs",
-                    # There's a bug in the ordering of outputs when there's a mapped `None` output
-                    # We have to stick with the deprecated API for now
-                    return_updates=True,
-                )
+            rng = pytensor.shared(np.random.default_rng())
+            xs, next_rng = scan(
+                fn=scan_step,
+                sequences=[
+                    pt.as_tensor_variable(np.array([-4, -3])),
+                    pt.as_tensor_variable(np.array([-2, -1])),
+                ],
+                outputs_info=[None, rng],
+                name="xs",
+                # There's a bug in the ordering of outputs when there's a mapped `None` output
+                # We have to stick with the deprecated API for now
+                return_updates=False,
+            )
             return xs
 
         with Model() as model:
@@ -674,22 +675,21 @@ def test_chained_custom_dist_bug(self):
         batch = 2
 
         def scan_dist(seq, n_steps, size):
-            def step(s):
-                innov = Normal.dist()
+            rng = pytensor.shared(np.random.default_rng())
+
+            def step(s, rng):
+                next_rng, innov = Normal.dist(rng=rng).owner.outputs
                 traffic = s + innov
-                return traffic, {innov.owner.inputs[0]: innov.owner.outputs[0]}
-
-            with pytest.warns(DeprecationWarning, match="Scan return signature will change"):
-                rv_seq, _ = pytensor.scan(
-                    fn=step,
-                    sequences=[seq],
-                    outputs_info=[None],
-                    n_steps=n_steps,
-                    strict=True,
-                    # There's a bug in the ordering of outputs when there's a mapped `None` output
-                    # We have to stick with the deprecated API for now
-                    return_updates=True,
-                )
+                return traffic, next_rng
+
+            rv_seq, _next_rng = pytensor.scan(
+                fn=step,
+                sequences=[seq],
+                outputs_info=[None, rng],
+                n_steps=n_steps,
+                strict=True,
+                return_updates=False,
+            )
             return rv_seq
 
         def normal_shifted(mu, size):