Remove numpy restriction (#990)

* Remove numpy restriction * fix pickling error get_potential_fn.<locals>.postprocess_fn * change file * Update helpers.py * pylint fix
arviz-devs · Jan 24, 2020 · ad5819c · ad5819c
1 parent 285ed3d
commit ad5819c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/arviz/tests/helpers.py b/arviz/tests/helpers.py
@@ -324,7 +324,8 @@ def numpyro_schools_model(data, draws, chains):
     # This block lets the posterior be pickled
     mcmc.sampler._sample_fn = None  # pylint: disable=protected-access
     mcmc.sampler._init_fn = None  # pylint: disable=protected-access
-    mcmc.sampler._constrain_fn = None  # pylint: disable=protected-access
+    mcmc.sampler._postprocess_fn = None  # pylint: disable=protected-access
+    mcmc.sampler._potential_fn = None  # pylint: disable=protected-access
     mcmc._cache = {}  # pylint: disable=protected-access
     return mcmc
 

diff --git a/requirements.txt b/requirements.txt
@@ -1,5 +1,5 @@
 matplotlib>=3.0
-numpy>=1.12,<1.18
+numpy>=1.12
 scipy>=0.19
 packaging
 pandas>=0.23