Bump 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.31.7,<2.32
+- pytensor>=2.32.0,<2.33
 - 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.31.7,<2.32
+- pytensor>=2.32.0,<2.33
 - 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.31.7,<2.32
+- pytensor>=2.32.0,<2.33
 - 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.31.7,<2.32
+- pytensor>=2.32.0,<2.33
 - 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.31.7,<2.32
+- pytensor>=2.32.0,<2.33
 - 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.31.7,<2.32
+- pytensor>=2.32.0,<2.33
 - python-graphviz
 - networkx
 - rich>=13.7.1

pymc/distributions/continuous.py

@@ -1618,7 +1618,7 @@ def support_point(rv, size, b, kappa, mu):
 
     def logp(value, b, kappa, mu):
         value = value - mu
-        res = pt.log(b / (kappa + (kappa**-1))) + (
+        res = pt.log(b / (kappa + pt.reciprocal(kappa))) + (
             -value * b * pt.sign(value) * (kappa ** pt.sign(value))
         )
 

pymc/logprob/linalg.py

@@ -14,7 +14,7 @@
 import pytensor.tensor as pt
 
 from pytensor.graph.rewriting.basic import node_rewriter
-from pytensor.tensor.math import _matrix_matrix_matmul
+from pytensor.tensor.math import _matmul
 
 from pymc.logprob.abstract import MeasurableBlockwise, MeasurableOp, _logprob, _logprob_helper
 from pymc.logprob.rewriting import measurable_ir_rewrites_db
@@ -55,7 +55,7 @@ def logprob_measurable_matmul(op, values, l, r):  # noqa: E741
     return x_logp - log_abs_jac_det
 
 
-@node_rewriter(tracks=[_matrix_matrix_matmul])
+@node_rewriter(tracks=[_matmul])
 def find_measurable_matmul(fgraph, node):
     """Find measurable matrix-matrix multiplication operations."""
     if isinstance(node.op, MeasurableOp):

pymc/ode/utils.py

@@ -83,7 +83,7 @@ def augment_system(ode_func, n_states, n_theta):
         Augemted system of differential equations.
     """
     # Present state of the system
-    t_y = pt.vector("y", dtype="float64")
+    t_y = pt.vector("y", dtype="float64", shape=(n_states,))
     t_y.tag.test_value = np.ones((n_states,), dtype="float64")
     # Parameter(s).  Should be vector to allow for generaliztion to multiparameter
     # systems of ODEs.  Is m dimensional because it includes all initial conditions as well as ode parameters

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.31.7,<2.32
+pytensor>=2.32.0,<2.33
 pytest-cov>=2.5
 pytest>=3.0
 rich>=13.7.1