|
43 | 43 |
|
44 | 44 | from pymc.aesaraf import change_rv_size, floatX, intX |
45 | 45 | from pymc.distributions import transforms |
46 | | -from pymc.distributions.continuous import ( |
47 | | - BoundedContinuous, |
48 | | - ChiSquared, |
49 | | - Normal, |
50 | | - assert_negative_support, |
51 | | -) |
| 46 | +from pymc.distributions.continuous import BoundedContinuous, ChiSquared, Normal |
52 | 47 | from pymc.distributions.dist_math import ( |
53 | 48 | betaln, |
54 | 49 | check_parameters, |
@@ -381,7 +376,7 @@ def dist(cls, nu, Sigma=None, mu=None, cov=None, tau=None, chol=None, lower=True |
381 | 376 | cov = quaddist_matrix(cov, chol, tau, lower) |
382 | 377 | # Aesara is stricter about the shape of mu, than PyMC used to be |
383 | 378 | mu = at.broadcast_arrays(mu, cov[..., -1])[0] |
384 | | - assert_negative_support(nu, "nu", "MvStudentT") |
| 379 | + |
385 | 380 | return super().dist([nu, mu, cov], **kwargs) |
386 | 381 |
|
387 | 382 | def moment(rv, size, nu, mu, cov): |
@@ -2288,9 +2283,6 @@ def dist(cls, alpha, K, *args, **kwargs): |
2288 | 2283 | alpha = at.as_tensor_variable(floatX(alpha)) |
2289 | 2284 | K = at.as_tensor_variable(intX(K)) |
2290 | 2285 |
|
2291 | | - assert_negative_support(alpha, "alpha", "StickBreakingWeights") |
2292 | | - assert_negative_support(K, "K", "StickBreakingWeights") |
2293 | | - |
2294 | 2286 | return super().dist([alpha, K], **kwargs) |
2295 | 2287 |
|
2296 | 2288 | def moment(rv, size, alpha, K): |
|
0 commit comments