From 21a3e97b2af128b5a2865f4653cd39ad1d30ec72 Mon Sep 17 00:00:00 2001 From: Ricardo Vieira Date: Mon, 7 Nov 2022 19:31:18 +0100 Subject: [PATCH] Use default Aesara RandomStream for VI --- pymc/aesaraf.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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,