Merge branch 'main' into llama

lm_eval/tasks/global_mmlu/default/pt/_pt_template_yaml

+dataset_path: CohereForAI/Global-MMLU-Lite
+dataset_name: pt
+test_split: test
+fewshot_split: dev
+fewshot_config:
+  sampler: default
+output_type: multiple_choice
+doc_to_text: "{{question.strip()}}\nA. {{option_a}}\nB. {{option_b}}\nC. {{option_c}}\nD. {{option_d}}\nAnswer:"
+doc_to_choice: ["A", "B", "C", "D"]
+doc_to_target: answer
+metric_list:
+  - metric: acc
+    aggregation: mean
+    higher_is_better: true
+metadata:
+  version: 0.0

lm_eval/tasks/global_mmlu/default/pt/global_mmlu_pt_business.yaml

+# Generated by _generate_configs.py
+include: _pt_template_yaml
+process_docs: !function utils.process_business
+task: global_mmlu_pt_business

lm_eval/tasks/global_mmlu/default/pt/global_mmlu_pt_humanities.yaml

+# Generated by _generate_configs.py
+include: _pt_template_yaml
+process_docs: !function utils.process_humanities
+task: global_mmlu_pt_humanities

lm_eval/tasks/global_mmlu/default/pt/global_mmlu_pt_medical.yaml

+# Generated by _generate_configs.py
+include: _pt_template_yaml
+process_docs: !function utils.process_medical
+task: global_mmlu_pt_medical

lm_eval/tasks/global_mmlu/default/pt/global_mmlu_pt_other.yaml

+# Generated by _generate_configs.py
+include: _pt_template_yaml
+process_docs: !function utils.process_other
+task: global_mmlu_pt_other

lm_eval/tasks/global_mmlu/default/pt/global_mmlu_pt_social_sciences.yaml

+# Generated by _generate_configs.py
+include: _pt_template_yaml
+process_docs: !function utils.process_social_sciences
+task: global_mmlu_pt_social_sciences

lm_eval/tasks/global_mmlu/default/pt/global_mmlu_pt_stem.yaml

+# Generated by _generate_configs.py
+include: _pt_template_yaml
+process_docs: !function utils.process_stem
+task: global_mmlu_pt_stem

lm_eval/tasks/global_mmlu/default/pt/utils.py

+from functools import partial
+
+
+CATEGORIES = ["Business", "Humanities", "Medical", "Other", "STEM", "Social Sciences"]
+
+
+def process_docs(dataset, category):
+    return dataset.filter(lambda x: x["subject_category"] == category)
+
+
+process_functions = {
+    f"process_{category.lower().replace(' ', '_')}": partial(
+        process_docs, category=category
+    )
+    for category in CATEGORIES
+}
+
+globals().update(process_functions)

lm_eval/tasks/global_mmlu/default/sw/_global_mmlu_sw.yaml

+group: global_mmlu_sw
+task:
+  - global_mmlu_sw_business
+  - global_mmlu_sw_humanities
+  - global_mmlu_sw_medical
+  - global_mmlu_sw_other
+  - global_mmlu_sw_stem
+  - global_mmlu_sw_social_sciences
+aggregate_metric_list:
+  - metric: acc
+    weight_by_size: True
+metadata:
+  version: 0.0

lm_eval/tasks/global_mmlu/default/sw/_sw_template_yaml

+dataset_path: CohereForAI/Global-MMLU-Lite
+dataset_name: sw
+test_split: test
+fewshot_split: dev
+fewshot_config:
+  sampler: default
+output_type: multiple_choice
+doc_to_text: "{{question.strip()}}\nA. {{option_a}}\nB. {{option_b}}\nC. {{option_c}}\nD. {{option_d}}\nAnswer:"
+doc_to_choice: ["A", "B", "C", "D"]
+doc_to_target: answer
+metric_list:
+  - metric: acc
+    aggregation: mean
+    higher_is_better: true
+metadata:
+  version: 0.0

lm_eval/tasks/global_mmlu/default/sw/global_mmlu_sw_business.yaml

+# Generated by _generate_configs.py
+include: _sw_template_yaml
+process_docs: !function utils.process_business
+task: global_mmlu_sw_business

lm_eval/tasks/global_mmlu/default/sw/global_mmlu_sw_humanities.yaml

+# Generated by _generate_configs.py
+include: _sw_template_yaml
+process_docs: !function utils.process_humanities
+task: global_mmlu_sw_humanities

lm_eval/tasks/global_mmlu/default/sw/global_mmlu_sw_medical.yaml

+# Generated by _generate_configs.py
+include: _sw_template_yaml
+process_docs: !function utils.process_medical
+task: global_mmlu_sw_medical

lm_eval/tasks/global_mmlu/default/sw/global_mmlu_sw_other.yaml

+# Generated by _generate_configs.py
+include: _sw_template_yaml
+process_docs: !function utils.process_other
+task: global_mmlu_sw_other

lm_eval/tasks/global_mmlu/default/sw/global_mmlu_sw_social_sciences.yaml

+# Generated by _generate_configs.py
+include: _sw_template_yaml
+process_docs: !function utils.process_social_sciences
+task: global_mmlu_sw_social_sciences

lm_eval/tasks/global_mmlu/default/sw/global_mmlu_sw_stem.yaml

+# Generated by _generate_configs.py
+include: _sw_template_yaml
+process_docs: !function utils.process_stem
+task: global_mmlu_sw_stem

lm_eval/tasks/global_mmlu/default/sw/utils.py

+from functools import partial
+
+
+CATEGORIES = ["Business", "Humanities", "Medical", "Other", "STEM", "Social Sciences"]
+
+
+def process_docs(dataset, category):
+    return dataset.filter(lambda x: x["subject_category"] == category)
+
+
+process_functions = {
+    f"process_{category.lower().replace(' ', '_')}": partial(
+        process_docs, category=category
+    )
+    for category in CATEGORIES
+}
+
+globals().update(process_functions)

lm_eval/tasks/global_mmlu/default/yo/_global_mmlu_yo.yaml

+group: global_mmlu_yo
+task:
+  - global_mmlu_yo_business
+  - global_mmlu_yo_humanities
+  - global_mmlu_yo_medical
+  - global_mmlu_yo_other
+  - global_mmlu_yo_stem
+  - global_mmlu_yo_social_sciences
+aggregate_metric_list:
+  - metric: acc
+    weight_by_size: True
+metadata:
+  version: 0.0

lm_eval/tasks/global_mmlu/default/yo/_yo_template_yaml

+dataset_path: CohereForAI/Global-MMLU-Lite
+dataset_name: yo
+test_split: test
+fewshot_split: dev
+fewshot_config:
+  sampler: default
+output_type: multiple_choice
+doc_to_text: "{{question.strip()}}\nA. {{option_a}}\nB. {{option_b}}\nC. {{option_c}}\nD. {{option_d}}\nAnswer:"
+doc_to_choice: ["A", "B", "C", "D"]
+doc_to_target: answer
+metric_list:
+  - metric: acc
+    aggregation: mean
+    higher_is_better: true
+metadata:
+  version: 0.0

lm_eval/tasks/global_mmlu/default/yo/global_mmlu_yo_business.yaml

+# Generated by _generate_configs.py
+include: _yo_template_yaml
+process_docs: !function utils.process_business
+task: global_mmlu_yo_business