Skip to content

Commit 541afd8

Browse files
Merge pull request #2243 from AustinRochford/mixture-bound-dont-broadcast-conditions
Don't broadcast conditions in Mixture logp
2 parents 9327794 + 9047dfa commit 541afd8

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

pymc3/distributions/mixture.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -111,7 +111,8 @@ def logp(self, value):
111111
w = self.w
112112

113113
return bound(logsumexp(tt.log(w) + self._comp_logp(value), axis=-1).sum(),
114-
w >= 0, w <= 1, tt.allclose(w.sum(axis=-1), 1))
114+
w >= 0, w <= 1, tt.allclose(w.sum(axis=-1), 1),
115+
broadcast_conditions=False)
115116

116117
def random(self, point=None, size=None, repeat=None):
117118
def random_choice(*args, **kwargs):

0 commit comments

Comments
 (0)