diff --git a/pymc/aesaraf.py b/pymc/aesaraf.py index a22fa5abc9..0835e20104 100644 --- a/pymc/aesaraf.py +++ b/pymc/aesaraf.py @@ -46,10 +46,10 @@ ) from aesara.graph.fg import FunctionGraph from aesara.graph.op import Op -from aesara.sandbox.rng_mrg import MRG_RandomStream as RandomStream from aesara.scalar.basic import Cast from aesara.tensor.basic import _as_tensor_variable from aesara.tensor.elemwise import Elemwise +from aesara.tensor.random import RandomStream from aesara.tensor.random.op import RandomVariable from aesara.tensor.random.var import ( RandomGeneratorSharedVariable,