diff --git a/graphicle/select.py b/graphicle/select.py index e828ea6..4aeb71c 100644 --- a/graphicle/select.py +++ b/graphicle/select.py @@ -1376,6 +1376,11 @@ def monte_carlo_tag( if clustered_pmu is None: clustered_pmu = particles.pmu[particles.final] ref_length = "particles.final" + if len(cluster_masks) < len(hard_pmu): + raise ValueError( + f"shape mismatch: only {len(cluster_masks)} clusters " + f"passed to tag {len(hard_pmu)} partons." + ) if len(clustered_pmu) != len(cluster_masks[0]): raise ValueError( "shape mismatch: length of elements in cluster_masks must be the "