Skip to content

Commit 2feff5e

Browse files
ArmavicaricardoV94
authored andcommitted
Fix opt -> rewriting deprecations from aesara
1 parent 7f2b87e commit 2feff5e

File tree

3 files changed

+4
-4
lines changed

3 files changed

+4
-4
lines changed

pymc/distributions/timeseries.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,9 +27,9 @@
2727
from aesara.graph.basic import Node
2828
from aesara.raise_op import Assert
2929
from aesara.tensor import TensorVariable
30-
from aesara.tensor.basic_opt import ShapeFeature, topo_constant_folding
3130
from aesara.tensor.random.op import RandomVariable
3231
from aesara.tensor.random.utils import normalize_size_param
32+
from aesara.tensor.rewriting.basic import ShapeFeature, topo_constant_folding
3333

3434
from pymc.aesaraf import change_rv_size, convert_observed_data, floatX, intX
3535
from pymc.distributions import distribution, multivariate
@@ -495,7 +495,7 @@ def _get_ar_order(cls, rhos: TensorVariable, ar_order: Optional[int], constant:
495495
features=[ShapeFeature()],
496496
clone=True,
497497
)
498-
(folded_shape,) = rewrite_graph(shape_fg, custom_opt=topo_constant_folding).outputs
498+
(folded_shape,) = rewrite_graph(shape_fg, custom_rewrite=topo_constant_folding).outputs
499499
folded_shape = getattr(folded_shape, "data", None)
500500
if folded_shape is None:
501501
raise ValueError(

pymc/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@
4343
from aesara.compile.sharedvalue import SharedVariable
4444
from aesara.graph.basic import Constant, Variable, graph_inputs
4545
from aesara.graph.fg import FunctionGraph
46-
from aesara.tensor.random.opt import local_subtensor_rv_lift
46+
from aesara.tensor.random.rewriting import local_subtensor_rv_lift
4747
from aesara.tensor.sharedvar import ScalarSharedVariable
4848
from aesara.tensor.var import TensorConstant, TensorVariable
4949

pymc/tests/helpers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121
import numpy.random as nr
2222

2323
from aesara.gradient import verify_grad as at_verify_grad
24-
from aesara.graph.opt import in2out
24+
from aesara.graph.rewriting.basic import in2out
2525
from aesara.sandbox.rng_mrg import MRG_RandomStream as RandomStream
2626

2727
from pymc.aesaraf import at_rng, local_check_parameter_to_ninf_switch, set_at_rng

0 commit comments

Comments
 (0)