Update PyTensor dependency

Co-authored-by: Jesse Grabowski <48652735+jessegrabowski@users.noreply.github.com>

Author: ricardoV94

conda-envs/environment-alternative-backends.yml

@@ -22,7 +22,7 @@ dependencies:
 - numpyro>=0.8.0
 - pandas>=0.24.0
 - pip
-- pytensor>=2.32.0,<2.33
+- pytensor>=2.34.0,<2.35
 - 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.32.0,<2.33
+- pytensor>=2.34.0,<2.35
 - 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.32.0,<2.33
+- pytensor>=2.34.0,<2.35
 - 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.32.0,<2.33
+- pytensor>=2.34.0,<2.35
 - 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.32.0,<2.33
+- pytensor>=2.34.0,<2.35
 - 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.32.0,<2.33
+- pytensor>=2.34.0,<2.35
 - python-graphviz
 - networkx
 - rich>=13.7.1

pymc/distributions/custom.py

@@ -18,9 +18,9 @@
 
 from pytensor import Variable, clone_replace
 from pytensor import tensor as pt
-from pytensor.graph.basic import io_toposort
 from pytensor.graph.features import ReplaceValidate
 from pytensor.graph.rewriting.basic import GraphRewriter
+from pytensor.graph.traversal import io_toposort
 from pytensor.scan.op import Scan
 from pytensor.tensor import TensorVariable, as_tensor_variable
 from pytensor.tensor.random.op import RandomVariable

pymc/distributions/timeseries.py

@@ -21,8 +21,9 @@
 import pytensor
 import pytensor.tensor as pt
 
-from pytensor.graph.basic import Apply, ancestors
+from pytensor.graph.basic import Apply
 from pytensor.graph.replace import clone_replace
+from pytensor.graph.traversal import ancestors
 from pytensor.tensor import TensorVariable
 from pytensor.tensor.random.op import RandomVariable
 from pytensor.tensor.random.utils import normalize_size_param

pymc/initial_point.py

@@ -22,9 +22,10 @@
 
 from pytensor import graph_replace
 from pytensor.compile.ops import TypeCastingOp
-from pytensor.graph.basic import Apply, Variable, ancestors, walk
+from pytensor.graph.basic import Apply, Variable
 from pytensor.graph.fg import FunctionGraph
 from pytensor.graph.rewriting.db import RewriteDatabaseQuery, SequenceDB
+from pytensor.graph.traversal import ancestors, walk
 from pytensor.tensor.variable import TensorVariable
 
 from pymc.logprob.transforms import Transform

pymc/logprob/basic.py

@@ -45,10 +45,9 @@
 from pytensor.graph.basic import (
     Constant,
     Variable,
-    ancestors,
-    walk,
 )
 from pytensor.graph.rewriting.basic import GraphRewriter, NodeRewriter
+from pytensor.graph.traversal import ancestors, walk
 from pytensor.tensor.variable import TensorVariable
 
 from pymc.logprob.abstract import (
@@ -533,8 +532,8 @@ def conditional_logp(
             f"The logprob terms of the following value variables could not be derived: {missing_value_terms}"
         )
 
-    values, logprobs = zip(*values_to_logprobs.items())
-    logprobs = cleanup_ir(logprobs)
+    # Ensure same order as input
+    logprobs = cleanup_ir(tuple(values_to_logprobs[v] for v in original_values))
 
     if warn_rvs:
         rvs_in_logp_expressions = _find_unallowed_rvs_in_graph(logprobs)
@@ -545,7 +544,7 @@ def conditional_logp(
                 UserWarning,
             )
 
-    return dict(zip(values, logprobs))
+    return dict(zip(original_values, logprobs))
 
 
 def transformed_conditional_logp(