Update Aesara dependency and add regression test for #5090

pymc-devs · Nov 3, 2021 · e52b2f7 · e52b2f7
1 parent 8a27ff4
commit e52b2f7
Show file tree

Hide file tree

Showing 11 changed files with 20 additions and 10 deletions.
diff --git a/conda-envs/environment-dev-py37.yml b/conda-envs/environment-dev-py37.yml
@@ -5,7 +5,7 @@ channels:
 - defaults
 dependencies:
 - aeppl>=0.0.13
-- aesara>=2.2.2
+- aesara>=2.2.6
 - arviz>=0.11.4
 - cachetools>=4.2.1
 - cloudpickle

diff --git a/conda-envs/environment-dev-py38.yml b/conda-envs/environment-dev-py38.yml
@@ -5,7 +5,7 @@ channels:
 - defaults
 dependencies:
 - aeppl>=0.0.13
-- aesara>=2.2.2
+- aesara>=2.2.6
 - arviz>=0.11.4
 - cachetools>=4.2.1
 - cloudpickle

diff --git a/conda-envs/environment-dev-py39.yml b/conda-envs/environment-dev-py39.yml
@@ -5,7 +5,7 @@ channels:
 - defaults
 dependencies:
 - aeppl>=0.0.13
-- aesara>=2.2.2
+- aesara>=2.2.6
 - arviz>=0.11.4
 - cachetools>=4.2.1
 - cloudpickle

diff --git a/conda-envs/environment-test-py37.yml b/conda-envs/environment-test-py37.yml
@@ -5,7 +5,7 @@ channels:
 - defaults
 dependencies:
 - aeppl>=0.0.13
-- aesara>=2.2.2
+- aesara>=2.2.6
 - arviz>=0.11.4
 - cachetools>=4.2.1
 - cloudpickle

diff --git a/conda-envs/environment-test-py38.yml b/conda-envs/environment-test-py38.yml
@@ -5,7 +5,7 @@ channels:
 - defaults
 dependencies:
 - aeppl>=0.0.13
-- aesara>=2.2.2
+- aesara>=2.2.6
 - arviz>=0.11.4
 - cachetools>=4.2.1
 - cloudpickle

diff --git a/conda-envs/environment-test-py39.yml b/conda-envs/environment-test-py39.yml
@@ -5,7 +5,7 @@ channels:
 - defaults
 dependencies:
 - aeppl>=0.0.13
-- aesara>=2.2.2
+- aesara>=2.2.6
 - arviz>=0.11.4
 - cachetools
 - cloudpickle

diff --git a/conda-envs/windows-environment-dev-py38.yml b/conda-envs/windows-environment-dev-py38.yml
@@ -5,7 +5,7 @@ channels:
 dependencies:
  # base dependencies (see install guide for Windows)
 - aeppl>=0.0.13
-- aesara>=2.2.2
+- aesara>=2.2.6
 - arviz>=0.11.4
 - cachetools>=4.2.1
 - cloudpickle

diff --git a/conda-envs/windows-environment-test-py38.yml b/conda-envs/windows-environment-test-py38.yml
@@ -5,7 +5,7 @@ channels:
 dependencies:
  # base dependencies (see install guide for Windows)
 - aeppl>=0.0.13
-- aesara>=2.2.2
+- aesara>=2.2.6
 - arviz>=0.11.2
 - cachetools
 - cloudpickle

diff --git a/pymc/tests/test_initial_point.py b/pymc/tests/test_initial_point.py
@@ -13,6 +13,7 @@
 #   limitations under the License.
 import aesara
 import aesara.tensor as at
+import cloudpickle
 import numpy as np
 import pytest
 
@@ -218,3 +219,12 @@ def test_moment_from_dims(self, rv_cls):
             assert not hasattr(rv.tag, "test_value")
             assert tuple(get_moment(rv).shape.eval()) == (4, 3)
         pass
+
+
+def test_pickling_issue_5090():
+    with pm.Model() as model:
+        pm.Normal("x", initval="prior")
+    ip_before = model.recompute_initial_point(seed=5090)
+    model = cloudpickle.loads(cloudpickle.dumps(model))
+    ip_after = model.recompute_initial_point(seed=5090)
+    assert ip_before["x"] == ip_after["x"]
diff --git a/requirements-dev.txt b/requirements-dev.txt
@@ -2,7 +2,7 @@
 # See that file for comments about the need/usage of each dependency.
 
 aeppl>=0.0.13
-aesara>=2.2.2
+aesara>=2.2.6
 arviz>=0.11.4
 cachetools>=4.2.1
 cloudpickle

diff --git a/requirements.txt b/requirements.txt
@@ -1,5 +1,5 @@
 aeppl>=0.0.13
-aesara>=2.2.2
+aesara>=2.2.6
 arviz>=0.11.4
 cachetools>=4.2.1
 cloudpickle