diff --git a/mcbackend/adapters/pymc.py b/mcbackend/adapters/pymc.py index 1f4e25b..f6b4ab9 100644 --- a/mcbackend/adapters/pymc.py +++ b/mcbackend/adapters/pymc.py @@ -7,8 +7,14 @@ import hagelkorn import numpy -from aesara.graph.basic import Constant -from aesara.tensor.sharedvar import SharedVariable + +try: + from pytensor.graph.basic import Constant + from pytensor.tensor.sharedvar import SharedVariable +except ModuleNotFoundError: + from aesara.graph.basic import Constant + from aesara.tensor.sharedvar import SharedVariable + from pymc.backends.base import BaseTrace from pymc.model import Model diff --git a/requirements-dev.txt b/requirements-dev.txt index f092d91..c032a6b 100644 --- a/requirements-dev.txt +++ b/requirements-dev.txt @@ -1,10 +1,8 @@ arviz clickhouse-driver flake8 -pymc==4.4.0 +pymc==5.0.0 pytest pytest-cov -# Temporary pin to fix Aesara import -setuptools>=48.0.0 twine wheel