24602abe9b89c6b22f77804636de2f109f1c6052,sympy/stats/symbolic_multivariate_probability.py,ExpectationMatrix,expand,#ExpectationMatrix#,66

Before Change

            return expr

        if isinstance(expr, Add):
            return Add(*[Expectation(a, condition=condition).expand() for a in expr.args])
        elif isinstance(expr, (Mul, MatMul)):
            if isinstance(_expand(expr), Add):
                return Expectation(_expand(expr)).expand()

After Change

        expand_expr = _expand(expr)
        if isinstance(expand_expr, Add):
            return Add.fromiter(Expectation(a, condition=condition).expand()
                    for a in expand_expr.args)

        elif isinstance(expr, (Mul, MatMul)):
            rv = []
            nonrv = []

Instances