cristobal-sifon / plottery

Plotting tools
GNU General Public License v3.0
0 stars 0 forks source link

corner crashes with chains of different size #19

Open cristobal-sifon opened 3 years ago

cristobal-sifon commented 3 years ago

if I pass a list of two chains with same number of parameters but different number of samples, it crashes with (in this case, the two chains have shapes ((10, 995), (10, 199)))

---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
<ipython-input-89-f82da7caad19> in <module>
     13     ls = ['-']
     14     names = ''
---> 15 fig, axes_diagonal, axes_offdiag = corner(
     16     p, labels=labels, limits=limits, blind=True, ls1d=ls, names=names,
     17     bins=10, bins1d=25, style1d='hist', background='density', bcolor='YlGnBu',

~/anaconda3/envs/kidsggl/lib/python3.8/site-packages/plottery/statsplots.py in corner(X, config, names, labels, bins, bins1d, clevels, contour_reference, blind, truths, truths_in_1d, truth_color, smooth, likelihood, likesmooth, color_likelihood, colors, cmap, ls1d, ls2d, style1d, medians1d, percentiles1d, background, bweight, bcolor, alpha, limits, show_likelihood_1d, ticks, show_contour, top_labels, pad, h_pad, w_pad, output, verbose, names_kwargs, contour_kwargs)
    205     # not yet implemented
    206     #options = _load_corner_config(config)
--> 207     nchains = (len(X) if _depth(X) > 2 else 1)
    208     if nchains > 1:
    209         ndim = len(X[0])

~/anaconda3/envs/kidsggl/lib/python3.8/site-packages/plottery/statsplots.py in _depth(L)
    562     scalar.
    563     """
--> 564     return len(np.array(L).shape)
    565 

ValueError: could not broadcast input array from shape (10,199) into shape (10,)