diff --git a/conda-envs/environment-alternative-backends.yml b/conda-envs/environment-alternative-backends.yml index b51ee3308d..1b575a1994 100644 --- a/conda-envs/environment-alternative-backends.yml +++ b/conda-envs/environment-alternative-backends.yml @@ -22,7 +22,7 @@ dependencies: - numpyro>=0.8.0 - pandas>=0.24.0 - pip -- pytensor>=2.28.3,<2.29 +- pytensor>=2.30.2,<2.31 - python-graphviz - networkx - rich>=13.7.1 diff --git a/conda-envs/environment-dev.yml b/conda-envs/environment-dev.yml index 857f7ef513..24da9d3966 100644 --- a/conda-envs/environment-dev.yml +++ b/conda-envs/environment-dev.yml @@ -12,7 +12,7 @@ dependencies: - numpy>=1.25.0 - pandas>=0.24.0 - pip -- pytensor>=2.28.3,<2.29 +- pytensor>=2.30.2,<2.31 - python-graphviz - networkx - scipy>=1.4.1 diff --git a/conda-envs/environment-docs.yml b/conda-envs/environment-docs.yml index 267c77cf2a..339f3c73e9 100644 --- a/conda-envs/environment-docs.yml +++ b/conda-envs/environment-docs.yml @@ -11,7 +11,7 @@ dependencies: - numpy>=1.25.0 - pandas>=0.24.0 - pip -- pytensor>=2.28.3,<2.29 +- pytensor>=2.30.2,<2.31 - python-graphviz - rich>=13.7.1 - scipy>=1.4.1 diff --git a/conda-envs/environment-test.yml b/conda-envs/environment-test.yml index dc732cfc49..f855163a6e 100644 --- a/conda-envs/environment-test.yml +++ b/conda-envs/environment-test.yml @@ -14,7 +14,7 @@ dependencies: - pandas>=0.24.0 - pip - polyagamma -- pytensor>=2.28.3,<2.29 +- pytensor>=2.30.2,<2.31 - python-graphviz - networkx - rich>=13.7.1 diff --git a/conda-envs/windows-environment-dev.yml b/conda-envs/windows-environment-dev.yml index 989a0558c4..d3df2b8a46 100644 --- a/conda-envs/windows-environment-dev.yml +++ b/conda-envs/windows-environment-dev.yml @@ -12,7 +12,7 @@ dependencies: - numpy>=1.25.0 - pandas>=0.24.0 - pip -- pytensor>=2.28.3,<2.29 +- pytensor>=2.30.2,<2.31 - python-graphviz - networkx - rich>=13.7.1 diff --git a/conda-envs/windows-environment-test.yml b/conda-envs/windows-environment-test.yml index 9435b068b0..4d070f3f9a 100644 --- a/conda-envs/windows-environment-test.yml +++ b/conda-envs/windows-environment-test.yml @@ -15,7 +15,7 @@ dependencies: - pandas>=0.24.0 - pip - polyagamma -- pytensor>=2.28.3,<2.29 +- pytensor>=2.30.2,<2.31 - python-graphviz - networkx - rich>=13.7.1 diff --git a/pymc/variational/opvi.py b/pymc/variational/opvi.py index f5ef6a4205..0903d27758 100644 --- a/pymc/variational/opvi.py +++ b/pymc/variational/opvi.py @@ -63,7 +63,6 @@ from pytensor.graph.replace import graph_replace from pytensor.scalar.basic import identity as scalar_identity from pytensor.tensor.elemwise import Elemwise -from pytensor.tensor.shape import unbroadcast import pymc as pm @@ -1096,13 +1095,6 @@ def make_size_and_deterministic_replacements(self, s, d, more_replacements=None) """ initial = self._new_initial(s, d, more_replacements) initial = pt.specify_shape(initial, self.symbolic_initial.type.shape) - # The static shape of initial may be more precise than self.symbolic_initial, - # and reveal previously unknown broadcastable dimensions. We have to mask those again. - if initial.type.broadcastable != self.symbolic_initial.type.broadcastable: - unbroadcast_axes = ( - i for i, b in enumerate(self.symbolic_initial.type.broadcastable) if not b - ) - initial = unbroadcast(initial, *unbroadcast_axes) if more_replacements: initial = graph_replace(initial, more_replacements, strict=False) return {self.symbolic_initial: initial} diff --git a/requirements-dev.txt b/requirements-dev.txt index becf5c7fcc..534503c249 100644 --- a/requirements-dev.txt +++ b/requirements-dev.txt @@ -16,7 +16,7 @@ numpydoc pandas>=0.24.0 polyagamma pre-commit>=2.8.0 -pytensor>=2.28.3,<2.29 +pytensor>=2.30.2,<2.31 pytest-cov>=2.5 pytest>=3.0 rich>=13.7.1 diff --git a/requirements.txt b/requirements.txt index c0a6151122..c278ad6917 100644 --- a/requirements.txt +++ b/requirements.txt @@ -3,7 +3,7 @@ cachetools>=4.2.1 cloudpickle numpy>=1.25.0 pandas>=0.24.0 -pytensor>=2.28.2,<2.29 +pytensor>=2.30.2,<2.31 rich>=13.7.1 scipy>=1.4.1 threadpoolctl>=3.1.0,<4.0.0