Merge branch 'main' into longcxt

# Conflicts:
#	lm_eval/tasks/README.md

lm_eval/tasks/global_mmlu/full/am/global_mmlu_full_am_medical_genetics.yaml

+# Generated by _generate_configs.py
+include: _am_template_yaml
+process_docs: !function utils.process_medical_genetics
+tag: global_mmlu_full_am_other_tasks
+task: global_mmlu_full_am_medical_genetics

lm_eval/tasks/global_mmlu/full/am/global_mmlu_full_am_miscellaneous.yaml

+# Generated by _generate_configs.py
+include: _am_template_yaml
+process_docs: !function utils.process_miscellaneous
+tag: global_mmlu_full_am_other_tasks
+task: global_mmlu_full_am_miscellaneous

lm_eval/tasks/global_mmlu/full/am/global_mmlu_full_am_moral_disputes.yaml

+# Generated by _generate_configs.py
+include: _am_template_yaml
+process_docs: !function utils.process_moral_disputes
+tag: global_mmlu_full_am_humanities_tasks
+task: global_mmlu_full_am_moral_disputes

lm_eval/tasks/global_mmlu/full/am/global_mmlu_full_am_moral_scenarios.yaml

+# Generated by _generate_configs.py
+include: _am_template_yaml
+process_docs: !function utils.process_moral_scenarios
+tag: global_mmlu_full_am_humanities_tasks
+task: global_mmlu_full_am_moral_scenarios

lm_eval/tasks/global_mmlu/full/am/global_mmlu_full_am_nutrition.yaml

+# Generated by _generate_configs.py
+include: _am_template_yaml
+process_docs: !function utils.process_nutrition
+tag: global_mmlu_full_am_other_tasks
+task: global_mmlu_full_am_nutrition

lm_eval/tasks/global_mmlu/full/am/global_mmlu_full_am_philosophy.yaml

+# Generated by _generate_configs.py
+include: _am_template_yaml
+process_docs: !function utils.process_philosophy
+tag: global_mmlu_full_am_humanities_tasks
+task: global_mmlu_full_am_philosophy

lm_eval/tasks/global_mmlu/full/am/global_mmlu_full_am_prehistory.yaml

+# Generated by _generate_configs.py
+include: _am_template_yaml
+process_docs: !function utils.process_prehistory
+tag: global_mmlu_full_am_humanities_tasks
+task: global_mmlu_full_am_prehistory

lm_eval/tasks/global_mmlu/full/am/global_mmlu_full_am_professional_accounting.yaml

+# Generated by _generate_configs.py
+include: _am_template_yaml
+process_docs: !function utils.process_professional_accounting
+tag: global_mmlu_full_am_other_tasks
+task: global_mmlu_full_am_professional_accounting

lm_eval/tasks/global_mmlu/full/am/global_mmlu_full_am_professional_law.yaml

+# Generated by _generate_configs.py
+include: _am_template_yaml
+process_docs: !function utils.process_professional_law
+tag: global_mmlu_full_am_humanities_tasks
+task: global_mmlu_full_am_professional_law

lm_eval/tasks/global_mmlu/full/am/global_mmlu_full_am_professional_medicine.yaml

+# Generated by _generate_configs.py
+include: _am_template_yaml
+process_docs: !function utils.process_professional_medicine
+tag: global_mmlu_full_am_other_tasks
+task: global_mmlu_full_am_professional_medicine

lm_eval/tasks/global_mmlu/full/am/global_mmlu_full_am_professional_psychology.yaml

+# Generated by _generate_configs.py
+include: _am_template_yaml
+process_docs: !function utils.process_professional_psychology
+tag: global_mmlu_full_am_social_sciences_tasks
+task: global_mmlu_full_am_professional_psychology

lm_eval/tasks/global_mmlu/full/am/global_mmlu_full_am_public_relations.yaml

+# Generated by _generate_configs.py
+include: _am_template_yaml
+process_docs: !function utils.process_public_relations
+tag: global_mmlu_full_am_social_sciences_tasks
+task: global_mmlu_full_am_public_relations

lm_eval/tasks/global_mmlu/full/am/global_mmlu_full_am_security_studies.yaml

+# Generated by _generate_configs.py
+include: _am_template_yaml
+process_docs: !function utils.process_security_studies
+tag: global_mmlu_full_am_social_sciences_tasks
+task: global_mmlu_full_am_security_studies

lm_eval/tasks/global_mmlu/full/am/global_mmlu_full_am_sociology.yaml

+# Generated by _generate_configs.py
+include: _am_template_yaml
+process_docs: !function utils.process_sociology
+tag: global_mmlu_full_am_social_sciences_tasks
+task: global_mmlu_full_am_sociology

lm_eval/tasks/global_mmlu/full/am/global_mmlu_full_am_us_foreign_policy.yaml

+# Generated by _generate_configs.py
+include: _am_template_yaml
+process_docs: !function utils.process_us_foreign_policy
+tag: global_mmlu_full_am_social_sciences_tasks
+task: global_mmlu_full_am_us_foreign_policy

lm_eval/tasks/global_mmlu/full/am/global_mmlu_full_am_virology.yaml

+# Generated by _generate_configs.py
+include: _am_template_yaml
+process_docs: !function utils.process_virology
+tag: global_mmlu_full_am_other_tasks
+task: global_mmlu_full_am_virology

lm_eval/tasks/global_mmlu/full/am/global_mmlu_full_am_world_religions.yaml

+# Generated by _generate_configs.py
+include: _am_template_yaml
+process_docs: !function utils.process_world_religions
+tag: global_mmlu_full_am_humanities_tasks
+task: global_mmlu_full_am_world_religions

lm_eval/tasks/global_mmlu/full/am/utils.py

+from functools import partial
+
+
+SUBJECTS = [
+    "abstract_algebra",
+    "anatomy",
+    "astronomy",
+    "business_ethics",
+    "clinical_knowledge",
+    "college_biology",
+    "college_chemistry",
+    "college_computer_science",
+    "college_mathematics",
+    "college_medicine",
+    "college_physics",
+    "computer_security",
+    "conceptual_physics",
+    "econometrics",
+    "electrical_engineering",
+    "elementary_mathematics",
+    "formal_logic",
+    "global_facts",
+    "high_school_biology",
+    "high_school_chemistry",
+    "high_school_computer_science",
+    "high_school_european_history",
+    "high_school_geography",
+    "high_school_government_and_politics",
+    "high_school_macroeconomics",
+    "high_school_mathematics",
+    "high_school_microeconomics",
+    "high_school_physics",
+    "high_school_psychology",
+    "high_school_statistics",
+    "high_school_us_history",
+    "high_school_world_history",
+    "human_aging",
+    "human_sexuality",
+    "international_law",
+    "jurisprudence",
+    "logical_fallacies",
+    "machine_learning",
+    "management",
+    "marketing",
+    "medical_genetics",
+    "miscellaneous",
+    "moral_disputes",
+    "moral_scenarios",
+    "nutrition",
+    "philosophy",
+    "prehistory",
+    "professional_accounting",
+    "professional_law",
+    "professional_medicine",
+    "professional_psychology",
+    "public_relations",
+    "security_studies",
+    "sociology",
+    "us_foreign_policy",
+    "virology",
+    "world_religions",
+]
+
+
+def process_docs(dataset, subject):
+    return dataset.filter(lambda x: x["subject"] == subject)
+
+
+process_functions = {
+    f"process_{subject}": partial(process_docs, subject=subject) for subject in SUBJECTS
+}
+
+globals().update(process_functions)

lm_eval/tasks/global_mmlu/full/ar/_ar_template_yaml

+dataset_path: CohereForAI/Global-MMLU
+dataset_name: ar
+test_split: test
+fewshot_split: dev
+fewshot_config:
+  sampler: first_n
+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/full/ar/_global_mmlu_full_ar.yaml

+group: global_mmlu_full_ar
+task:
+  - global_mmlu_full_ar_stem
+  - global_mmlu_full_ar_other
+  - global_mmlu_full_ar_social_sciences
+  - global_mmlu_full_ar_humanities
+aggregate_metric_list:
+  - metric: acc
+    weight_by_size: True
+metadata:
+  version: 0.0