Skip to content

Commit e9b9a44

Browse files
committed
Test for chol factorization should pass now
1 parent c5214af commit e9b9a44

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pymc3/tests/test_distributions.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1327,7 +1327,7 @@ def setup_class(self):
13271327
sigma = HalfNormal('sigma', sigma=1)
13281328

13291329
#Test Cholesky parameterization
1330-
Z = MvNormal('Z', mu = np.zeros(2), chol = np.eye(2), shape = 2)
1330+
Z = MvNormal('Z', mu=np.zeros(2), chol=np.eye(2), shape=(2,))
13311331

13321332
# Expected value of outcome
13331333
mu = Deterministic('mu', floatX(alpha + tt.dot(X, b)))
@@ -1340,7 +1340,7 @@ def setup_class(self):
13401340
r'$\text{sigma} \sim \text{HalfNormal}(\mathit{sigma}=1.0)$',
13411341
r'$\text{mu} \sim \text{Deterministic}(\text{alpha},~\text{Constant},~\text{beta})$',
13421342
r'$\text{beta} \sim \text{Normal}(\mathit{mu}=0.0,~\mathit{sigma}=10.0)$',
1343-
r'$\text{Z} \sim \text{MvNormal}(\mathit{mu}=array,~\mathit{chol}=array)$',
1343+
r'$Z \sim \text{MvNormal}(\mathit{mu}=array, \mathit{chol}=array)$',
13441344
r'$\text{Y_obs} \sim \text{Normal}(\mathit{mu}=\text{mu},~\mathit{sigma}=f(\text{sigma}))$'
13451345
)
13461346

0 commit comments

Comments
 (0)