From 6dafeae60f6016b6ddffc7d55f5191713d2a6460 Mon Sep 17 00:00:00 2001 From: "Brandon T. Willard" Date: Tue, 3 Jan 2023 13:56:10 -0600 Subject: [PATCH 1/2] Update AePPL version --- environment.yml | 4 ++-- setup.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/environment.yml b/environment.yml index 9e08eb9..5756eed 100644 --- a/environment.yml +++ b/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 diff --git a/setup.py b/setup.py index 477a508..a48ca02 100644 --- a/setup.py +++ b/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", From 0fad5ef4a1ae43c6c42653ee1b77dca20f085f59 Mon Sep 17 00:00:00 2001 From: "Brandon T. Willard" Date: Tue, 3 Jan 2023 13:54:36 -0600 Subject: [PATCH 2/2] Replace PreserveRVMappings with MeasurableConversionTracker --- aemcmc/rewriting.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/aemcmc/rewriting.py b/aemcmc/rewriting.py index 1971544..9039440 100644 --- a/aemcmc/rewriting.py +++ b/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