update imports

Author: williambdean

pymc_extras/model/marginal/graph_analysis.py

@@ -7,7 +7,7 @@
 from pymc.model.fgraph import ModelVar
 from pymc.variational.minibatch_rv import MinibatchRandomVariable
 from pytensor.graph import Variable, ancestors
-from pytensor.graph.basic import io_toposort
+from pytensor.graph.traversal import io_toposort
 from pytensor.tensor import TensorType, TensorVariable
 from pytensor.tensor.blockwise import Blockwise
 from pytensor.tensor.elemwise import CAReduce, DimShuffle, Elemwise

pymc_extras/sampling/optimizations/conjugacy.py

@@ -4,9 +4,10 @@
 from pymc.distributions import Beta, Binomial
 from pymc.model.fgraph import ModelFreeRV, ModelValuedVar, model_free_rv
 from pymc.pytensorf import collect_default_updates
-from pytensor.graph.basic import Variable, ancestors
+from pytensor.graph.basic import Variable
 from pytensor.graph.fg import FunctionGraph, Output
 from pytensor.graph.rewriting.basic import node_rewriter
+from pytensor.graph.traversal import ancestors
 from pytensor.tensor.elemwise import DimShuffle
 from pytensor.tensor.subtensor import _sum_grad_over_bcasted_dims as sum_bcasted_dims
 

pymc_extras/sampling/optimizations/conjugate_sampler.py

@@ -5,7 +5,7 @@
 from pymc.initial_point import PointType
 from pymc.logprob.abstract import MeasurableOp, _logprob
 from pymc.model.core import modelcontext
-from pymc.pytensorf import compile_pymc
+from pymc.pytensorf import compile
 from pymc.step_methods.compound import BlockedStep, Competence, StepMethodState
 from pymc.util import get_value_vars_from_user_vars
 from pytensor import shared
@@ -68,7 +68,7 @@ def __init__(self, vars, model=None, rng=None, compile_kwargs: dict | None = Non
 
         if compile_kwargs is None:
             compile_kwargs = {}
-        self.posterior_fn = compile_pymc(
+        self.posterior_fn = compile(
             model.value_vars,
             posterior_rv,
             random_seed=rng,