b63a01f72b06f4166c128ee2d543b89eed2c9b00,arviz/plots/densityplot.py,,densityplot,#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#,10

Before Change

        varnames = []
        for tr in trace:
            varnames_tmp = tr.columns
            for v in varnames_tmp:
                if v not in varnames:
                    varnames.append(v)
    else:
        v_tmp = []
        for tr in trace:
            v_tmp.extend(expand_variable_names(tr, varnames))

After Change

    if varnames is None:
        varnames = set.union(*[set(tr.columns) for tr in trace])
    else:
        varnames = set.union(*[set(expand_variable_names(tr, varnames)) for tr in trace])

    if figsize is None:
        figsize = (6, len(varnames) * 2)

Instances