aet -> at

Author: ferrine

pymc3/tests/test_variational_inference.py

@@ -964,7 +964,7 @@ def test_discrete_not_allowed():
 
     with pm.Model():
         mu = pm.Normal("mu", mu=0, sigma=10, size=3)
-        z = pm.Categorical("z", p=aet.ones(3) / 3, size=len(y))
+        z = pm.Categorical("z", p=at.ones(3) / 3, size=len(y))
         pm.Normal("y_obs", mu=mu[z], sigma=1.0, observed=y)
         with pytest.raises(opvi.ParametrizationError):
             pm.fit(n=1)  # fails