Skip to content

Commit 3e52058

Browse files
Take out size=1 special case from MixtureSameFamily
1 parent 25f6108 commit 3e52058

File tree

1 file changed

+0
-6
lines changed

1 file changed

+0
-6
lines changed

pymc3/distributions/mixture.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -870,12 +870,6 @@ def random(self, point=None, size=None):
870870

871871
# The `samples` array still has the `mixture_axis`, so we must remove it:
872872
output = samples[(..., 0) + (slice(None),) * len(event_shape)]
873-
874-
# Final oddity: if size == 1, pymc3 defaults to reducing the sample_shape dimension
875-
# We do this to stay consistent with the rest of the package even though
876-
# we shouldn't have to do it.
877-
if size == 1:
878-
output = output[0]
879873
return output
880874

881875
def _distr_parameters_for_repr(self):

0 commit comments

Comments
 (0)