throw error when tagging if not enough clusters #168

jacanchaplais · Nov 28, 2023 · dac2859 · dac2859
1 parent d30c5da
commit dac2859
Showing 1 changed file with 5 additions and 0 deletions.
diff --git 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 "