Update Aesara dependency and add regression test for #5090

Author: ricardoV94

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

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

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

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

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

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

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

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

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"]

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

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