diff --git a/src/toolbox_scs/routines/boz.py b/src/toolbox_scs/routines/boz.py index 4f1110f37344d6417b02595544a722cb4077ccd0..9c281c83540f990ce8d7d9fee33836c1d226d78a 100644 --- a/src/toolbox_scs/routines/boz.py +++ b/src/toolbox_scs/routines/boz.py @@ -524,9 +524,9 @@ def histogram_module(proposalNB, runNB, moduleNB, mask=None): w = da.repeat(da.repeat(da.array(mask[None, None, :, :]), arr.shape[1], axis=1), arr.shape[0], axis=0) w = w.rechunk((100, -1, -1, -1)) - return da.bincount(arr.ravel(), w.ravel()).compute() + return da.bincount(arr.ravel(), w.ravel(), minlength=512).compute() else: - return da.bincount(arr.ravel()).compute() + return da.bincount(arr.ravel(), minlength=512).compute() def inspect_histogram(proposalNB, runNB, moduleNB,