Bump PyTensor dependency

Author: ricardoV94

conda-envs/environment-dev.yml

@@ -13,7 +13,7 @@ dependencies:
 - numpy>=1.15.0
 - pandas>=0.24.0
 - pip
-- pytensor>=2.25.1,<2.26
+- pytensor>=2.26.2,<2.27
 - python-graphviz
 - networkx
 - scipy>=1.4.1

conda-envs/environment-docs.yml

@@ -11,7 +11,7 @@ dependencies:
 - numpy>=1.15.0
 - pandas>=0.24.0
 - pip
-- pytensor>=2.25.1,<2.26
+- pytensor>=2.26.2,<2.27
 - python-graphviz
 - rich>=13.7.1
 - scipy>=1.4.1

conda-envs/environment-jax.yml

@@ -20,7 +20,7 @@ dependencies:
 - numpyro>=0.8.0
 - pandas>=0.24.0
 - pip
-- pytensor>=2.25.1,<2.26
+- pytensor>=2.26.2,<2.27
 - python-graphviz
 - networkx
 - rich>=13.7.1

conda-envs/environment-test.yml

@@ -16,7 +16,7 @@ dependencies:
 - numpy>=1.15.0
 - pandas>=0.24.0
 - pip
-- pytensor>=2.25.1,<2.26
+- pytensor>=2.26.2,<2.27
 - python-graphviz
 - networkx
 - rich>=13.7.1

conda-envs/windows-environment-dev.yml

@@ -13,7 +13,7 @@ dependencies:
 - numpy>=1.15.0
 - pandas>=0.24.0
 - pip
-- pytensor>=2.25.1,<2.26
+- pytensor>=2.26.2,<2.27
 - python-graphviz
 - networkx
 - rich>=13.7.1

conda-envs/windows-environment-test.yml

@@ -16,7 +16,7 @@ dependencies:
 - numpy>=1.15.0
 - pandas>=0.24.0
 - pip
-- pytensor>=2.25.1,<2.26
+- pytensor>=2.26.2,<2.27
 - python-graphviz
 - networkx
 - rich>=13.7.1

pymc/logprob/tensor.py

@@ -286,9 +286,9 @@ def find_measurable_dimshuffles(fgraph, node) -> list[TensorVariable] | None:
 
     base_var = node.inputs[0]
 
-    measurable_dimshuffle = MeasurableDimShuffle(node.op.input_broadcastable, node.op.new_order)(
-        base_var
-    )
+    measurable_dimshuffle = MeasurableDimShuffle(
+        input_ndim=node.op.input_ndim, new_order=node.op.new_order
+    )(base_var)
     assert isinstance(measurable_dimshuffle, TensorVariable)
 
     return [measurable_dimshuffle]

requirements-dev.txt

@@ -17,7 +17,7 @@ numpydoc
 pandas>=0.24.0
 polyagamma
 pre-commit>=2.8.0
-pytensor>=2.25.1,<2.26
+pytensor>=2.26.2,<2.27
 pytest-cov>=2.5
 pytest>=3.0
 rich>=13.7.1

requirements.txt

@@ -3,7 +3,7 @@ cachetools>=4.2.1
 cloudpickle
 numpy>=1.15.0
 pandas>=0.24.0
-pytensor>=2.25.1,<2.26
+pytensor>=2.26.1,<2.27
 rich>=13.7.1
 scipy>=1.4.1
 threadpoolctl>=3.1.0,<4.0.0