Updates for new Aesara deprecations and AePPL version

Author: brandonwillard

aemcmc/transforms.py

@@ -113,7 +113,7 @@ def invgamma_exponential(invgamma_expr, invexponential_expr):
         size=size_lv,
         dtype=dtype_lv,
     )
-    invexponential_et = etuple(at.true_div, at.as_tensor(1.0), exponential_et)
+    invexponential_et = etuple(at.true_divide, at.as_tensor(1.0), exponential_et)
 
     return lall(
         eq(invgamma_expr, invgamma_et), eq(invexponential_expr, invexponential_et)

environment.yml

@@ -11,8 +11,8 @@ dependencies:
   - compilers
   - numpy>=1.18.1
   - scipy>=1.4.0
-  - aesara>=2.8.3
-  - aeppl>=0.1.0
+  - aesara>=2.8.11
+  - aeppl>=0.1.2
   - aehmc>=0.0.10
   - polyagamma>=1.3.2
   - cons

setup.py

@@ -38,8 +38,8 @@ def get_versions():
     install_requires=[
         "numpy>=1.18.1",
         "scipy>=1.4.0",
-        "aesara>=2.8.3",
-        "aeppl>=0.1.0",
+        "aesara>=2.8.11",
+        "aeppl>=0.1.2",
         "aehmc>=0.0.10",
         "polyagamma>=1.3.2",
         "cons",

tests/test_basic.py

@@ -145,7 +145,6 @@ def test_nuts_with_closed_form():
     assert beta_rv in sampler.sample_steps
 
 
-@pytest.mark.xfail(reason="An `OpFromGraph` fix is needed to remove extra updates")
 def test_create_gibbs():
     srng = RandomStream(0)
 

tests/test_conjugates.py

@@ -31,7 +31,6 @@ def test_gamma_poisson_conjugate_contract():
     q_lv = var()
     (posterior_expr,) = run(1, q_lv, gamma_poisson_conjugateo(y_vv, Y_rv, q_lv))
     posterior = eval_if_etuple(posterior_expr)
-    aesara.dprint(posterior)
 
     assert isinstance(posterior.owner.op, type(at.random.gamma))
 

tests/test_rewriting.py

@@ -119,7 +119,7 @@ def test_SubsumingElemwise_constant_inputs():
     srng = at.random.RandomStream(0)
 
     s = at.lscalar("s")
-    # The `1` is the constant input to a `true_div` `Elemwise` that should be
+    # The `1` is the constant input to a `true_divide` `Elemwise` that should be
     # "subsumed"
     Z = srng.exponential(1, size=s, name="Z")
     mu = 1 / Z