Test PyMC v5.0.0 compatibility

Author: michaelosthege

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
 

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