From 4ac3ca6bdf772a19677bfb2fcde54790eb61c93c Mon Sep 17 00:00:00 2001 From: Kyle Beauchamp Date: Sun, 21 Jun 2020 15:01:37 -0700 Subject: [PATCH 1/4] Add simple test case for coords behavior --- pymc3/tests/test_coords.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) create mode 100644 pymc3/tests/test_coords.py diff --git a/pymc3/tests/test_coords.py b/pymc3/tests/test_coords.py new file mode 100644 index 0000000000..2f7f24b320 --- /dev/null +++ b/pymc3/tests/test_coords.py @@ -0,0 +1,18 @@ +import pymc3 as pm +import numpy as np + + +def test_coords(): + chains = 2 + n_features = 3 + n_samples = 10 + + coords = {"features": np.arange(n_features)} + + with pm.Model(coords=coords): + a = pm.Uniform("a", -100, 100, dims="features") + b = pm.Uniform("b", -100, 100, dims="features") + tr = pm.sample(n_samples, chains=chains, return_inferencedata=True) + + assert "features" in tr.posterior.a.coords.dims + assert "features" in tr.posterior.b.coords.dims From 265cd328cf7117b663a660d8deed799fe30aa38c Mon Sep 17 00:00:00 2001 From: Kyle Beauchamp Date: Sun, 21 Jun 2020 17:17:42 -0700 Subject: [PATCH 2/4] Install arviz directly from github as temporary fix --- requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index ed4d8b812f..d5301d8b33 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ -arviz>=0.8.3 +-e git://github.com/arviz-devs/arviz.git#egg=arviz theano>=1.0.4 numpy>=1.13.0 scipy>=0.18.1 From a53a683cb63af44819f19ae6c7b1a67e4cb3ac9b Mon Sep 17 00:00:00 2001 From: Kyle Beauchamp Date: Sun, 21 Jun 2020 17:27:05 -0700 Subject: [PATCH 3/4] Revert attempt to point to latest arviz github --- requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index d5301d8b33..ed4d8b812f 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ --e git://github.com/arviz-devs/arviz.git#egg=arviz +arviz>=0.8.3 theano>=1.0.4 numpy>=1.13.0 scipy>=0.18.1 From c44af6d7f06daa676f734387c71634336cda792f Mon Sep 17 00:00:00 2001 From: Kyle Beauchamp Date: Tue, 23 Jun 2020 07:48:18 -0700 Subject: [PATCH 4/4] Bump arviz requirement --- requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index ed4d8b812f..4882d65218 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ -arviz>=0.8.3 +arviz>=0.9.0 theano>=1.0.4 numpy>=1.13.0 scipy>=0.18.1