Added grouping

Signed-off-by: Jonathan Bnayahu <bnayahu@il.ibm.com>

Author: bnayahu

prepare/benchmarks/safety.py

@@ -18,11 +18,13 @@
         "airbench": DatasetRecipe(
             card="cards.safety.airbench2024",
             template_card_index="default",
+            group_by=["l2-name"],
             max_test_instances=500,
         ),
         "ailuminate": DatasetRecipe(
             card="cards.safety.mlcommons_ailuminate",
             template_card_index="default",
+            group_by=["hazard"],
             max_test_instances=500,
         ),
     }

src/unitxt/catalog/benchmarks/safety.json

@@ -20,12 +20,18 @@
             "__type__": "dataset_recipe",
             "card": "cards.safety.airbench2024",
             "template_card_index": "default",
+            "group_by": [
+                "l2-name"
+            ],
             "max_test_instances": 500
         },
         "ailuminate": {
             "__type__": "dataset_recipe",
             "card": "cards.safety.mlcommons_ailuminate",
             "template_card_index": "default",
+            "group_by": [
+                "hazard"
+            ],
             "max_test_instances": 500
         }
     }