Updates for new valued/bound variables in AePPL

aemcmc/rewriting.py

@@ -2,7 +2,7 @@
 from functools import wraps
 from typing import Callable, Dict, Iterable, List, Optional, Sequence, Set, Tuple, Union
 
-from aeppl.rewriting import PreserveRVMappings
+from aeppl.rewriting import MeasurableConversionTracker
 from aesara.compile.builders import OpFromGraph
 from aesara.compile.mode import optdb
 from aesara.graph.basic import Apply, Variable, clone_replace, io_toposort
@@ -84,7 +84,7 @@ def construct_ir_fgraph(
         memo=memo,
         copy_orphans=False,
         copy_inputs=False,
-        features=[ShapeFeature(), PreserveRVMappings(obs_rvs_to_values)],
+        features=[ShapeFeature(), MeasurableConversionTracker()],
     )
 
     # Update `obs_rvs_to_values` so that it uses the new cloned variables

environment.yml

@@ -12,8 +12,8 @@ dependencies:
   - numpy>=1.18.1
   - scipy>=1.4.0
   - aesara>=2.8.3
-  - aeppl>=0.0.40
-  - aehmc>0.0.10
+  - aeppl>=0.1.0
+  - aehmc>=0.0.10
   - polyagamma>=1.3.2
   - cons
   - logical-unification

setup.py

@@ -39,8 +39,8 @@ def get_versions():
         "numpy>=1.18.1",
         "scipy>=1.4.0",
         "aesara>=2.8.3",
-        "aeppl>=0.0.40",
-        "aehmc>0.0.10",
+        "aeppl>=0.1.0",
+        "aehmc>=0.0.10",
         "polyagamma>=1.3.2",
         "cons",
         "logical-unification",