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