Skip to content

Commit

Permalink
Removed assert_negative_support function call
Browse files Browse the repository at this point in the history
  • Loading branch information
Willian Dihanster authored and michaelosthege committed Aug 8, 2022
1 parent 75e6be0 commit ad16bf4
Showing 1 changed file with 2 additions and 10 deletions.
12 changes: 2 additions & 10 deletions pymc/distributions/multivariate.py
Original file line number Diff line number Diff line change
Expand Up @@ -43,12 +43,7 @@

from pymc.aesaraf import change_rv_size, floatX, intX
from pymc.distributions import transforms
from pymc.distributions.continuous import (
BoundedContinuous,
ChiSquared,
Normal,
assert_negative_support,
)
from pymc.distributions.continuous import BoundedContinuous, ChiSquared, Normal
from pymc.distributions.dist_math import (
betaln,
check_parameters,
Expand Down Expand Up @@ -381,7 +376,7 @@ def dist(cls, nu, Sigma=None, mu=None, cov=None, tau=None, chol=None, lower=True
cov = quaddist_matrix(cov, chol, tau, lower)
# Aesara is stricter about the shape of mu, than PyMC used to be
mu = at.broadcast_arrays(mu, cov[..., -1])[0]
assert_negative_support(nu, "nu", "MvStudentT")

return super().dist([nu, mu, cov], **kwargs)

def moment(rv, size, nu, mu, cov):
Expand Down Expand Up @@ -2288,9 +2283,6 @@ def dist(cls, alpha, K, *args, **kwargs):
alpha = at.as_tensor_variable(floatX(alpha))
K = at.as_tensor_variable(intX(K))

assert_negative_support(alpha, "alpha", "StickBreakingWeights")
assert_negative_support(K, "K", "StickBreakingWeights")

return super().dist([alpha, K], **kwargs)

def moment(rv, size, alpha, K):
Expand Down

0 comments on commit ad16bf4

Please sign in to comment.