Skip to content

Commit b1c3164

Browse files
author
Chris Fonnesbeck
committed
Replaced AllocDiag class with alloc_diag function call
1 parent 3eff859 commit b1c3164

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pymc3/distributions/multivariate.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -488,15 +488,15 @@ class HWCov(Continuous):
488488
def __init__(self, nu, a, *args, **kwargs):
489489
self.nu = nu
490490
self.a = a
491-
self.mean = tt.nlinalg.AllocDiag(a)
491+
self.mean = tt.nlinalg.alloc_diag(a)
492492
super(HWCov, self).__init__(*args, **kwargs)
493493

494494
def random(self, point=None, size=None):
495495
nu, a = draw_values([self.nu, self.a], point=point)
496496

497497
def _random(nu, a, size=None):
498498
alpha = stats.invgamma.rvs(a=0.5, scale=1/(a**2))
499-
return stats.invwishart.rvs(df=nu+p-1, scale=2*nu*tt.nlinalg.AllocDiag(1/alpha))
499+
return stats.invwishart.rvs(df=nu+p-1, scale=2*nu*tt.nlinalg.alloc_diag(1/alpha))
500500

501501
samples = generate_samples(_random, nu, a,
502502
dist_shape=self.shape,

0 commit comments

Comments
 (0)