Merge branch 'main' into ai2d

lm_eval/tasks/global_mmlu/full/es/global_mmlu_full_es_management.yaml

+# Generated by _generate_configs.py
+include: _es_template_yaml
+process_docs: !function utils.process_management
+tag: global_mmlu_full_es_other_tasks
+task: global_mmlu_full_es_management

lm_eval/tasks/global_mmlu/full/es/global_mmlu_full_es_marketing.yaml

+# Generated by _generate_configs.py
+include: _es_template_yaml
+process_docs: !function utils.process_marketing
+tag: global_mmlu_full_es_other_tasks
+task: global_mmlu_full_es_marketing

lm_eval/tasks/global_mmlu/full/es/global_mmlu_full_es_medical_genetics.yaml

+# Generated by _generate_configs.py
+include: _es_template_yaml
+process_docs: !function utils.process_medical_genetics
+tag: global_mmlu_full_es_other_tasks
+task: global_mmlu_full_es_medical_genetics

lm_eval/tasks/global_mmlu/full/es/global_mmlu_full_es_miscellaneous.yaml

+# Generated by _generate_configs.py
+include: _es_template_yaml
+process_docs: !function utils.process_miscellaneous
+tag: global_mmlu_full_es_other_tasks
+task: global_mmlu_full_es_miscellaneous

lm_eval/tasks/global_mmlu/full/es/global_mmlu_full_es_moral_disputes.yaml

+# Generated by _generate_configs.py
+include: _es_template_yaml
+process_docs: !function utils.process_moral_disputes
+tag: global_mmlu_full_es_humanities_tasks
+task: global_mmlu_full_es_moral_disputes

lm_eval/tasks/global_mmlu/full/es/global_mmlu_full_es_moral_scenarios.yaml

+# Generated by _generate_configs.py
+include: _es_template_yaml
+process_docs: !function utils.process_moral_scenarios
+tag: global_mmlu_full_es_humanities_tasks
+task: global_mmlu_full_es_moral_scenarios

lm_eval/tasks/global_mmlu/full/es/global_mmlu_full_es_nutrition.yaml

+# Generated by _generate_configs.py
+include: _es_template_yaml
+process_docs: !function utils.process_nutrition
+tag: global_mmlu_full_es_other_tasks
+task: global_mmlu_full_es_nutrition

lm_eval/tasks/global_mmlu/full/es/global_mmlu_full_es_philosophy.yaml

+# Generated by _generate_configs.py
+include: _es_template_yaml
+process_docs: !function utils.process_philosophy
+tag: global_mmlu_full_es_humanities_tasks
+task: global_mmlu_full_es_philosophy

lm_eval/tasks/global_mmlu/full/es/global_mmlu_full_es_prehistory.yaml

+# Generated by _generate_configs.py
+include: _es_template_yaml
+process_docs: !function utils.process_prehistory
+tag: global_mmlu_full_es_humanities_tasks
+task: global_mmlu_full_es_prehistory

lm_eval/tasks/global_mmlu/full/es/global_mmlu_full_es_professional_accounting.yaml

+# Generated by _generate_configs.py
+include: _es_template_yaml
+process_docs: !function utils.process_professional_accounting
+tag: global_mmlu_full_es_other_tasks
+task: global_mmlu_full_es_professional_accounting

lm_eval/tasks/global_mmlu/full/es/global_mmlu_full_es_professional_law.yaml

+# Generated by _generate_configs.py
+include: _es_template_yaml
+process_docs: !function utils.process_professional_law
+tag: global_mmlu_full_es_humanities_tasks
+task: global_mmlu_full_es_professional_law

lm_eval/tasks/global_mmlu/full/es/global_mmlu_full_es_professional_medicine.yaml

+# Generated by _generate_configs.py
+include: _es_template_yaml
+process_docs: !function utils.process_professional_medicine
+tag: global_mmlu_full_es_other_tasks
+task: global_mmlu_full_es_professional_medicine

lm_eval/tasks/global_mmlu/full/es/global_mmlu_full_es_professional_psychology.yaml

+# Generated by _generate_configs.py
+include: _es_template_yaml
+process_docs: !function utils.process_professional_psychology
+tag: global_mmlu_full_es_social_sciences_tasks
+task: global_mmlu_full_es_professional_psychology

lm_eval/tasks/global_mmlu/full/es/global_mmlu_full_es_public_relations.yaml

+# Generated by _generate_configs.py
+include: _es_template_yaml
+process_docs: !function utils.process_public_relations
+tag: global_mmlu_full_es_social_sciences_tasks
+task: global_mmlu_full_es_public_relations

lm_eval/tasks/global_mmlu/full/es/global_mmlu_full_es_security_studies.yaml

+# Generated by _generate_configs.py
+include: _es_template_yaml
+process_docs: !function utils.process_security_studies
+tag: global_mmlu_full_es_social_sciences_tasks
+task: global_mmlu_full_es_security_studies

lm_eval/tasks/global_mmlu/full/es/global_mmlu_full_es_sociology.yaml

+# Generated by _generate_configs.py
+include: _es_template_yaml
+process_docs: !function utils.process_sociology
+tag: global_mmlu_full_es_social_sciences_tasks
+task: global_mmlu_full_es_sociology

lm_eval/tasks/global_mmlu/full/es/global_mmlu_full_es_us_foreign_policy.yaml

+# Generated by _generate_configs.py
+include: _es_template_yaml
+process_docs: !function utils.process_us_foreign_policy
+tag: global_mmlu_full_es_social_sciences_tasks
+task: global_mmlu_full_es_us_foreign_policy

lm_eval/tasks/global_mmlu/full/es/global_mmlu_full_es_virology.yaml

+# Generated by _generate_configs.py
+include: _es_template_yaml
+process_docs: !function utils.process_virology
+tag: global_mmlu_full_es_other_tasks
+task: global_mmlu_full_es_virology

lm_eval/tasks/global_mmlu/full/es/global_mmlu_full_es_world_religions.yaml

+# Generated by _generate_configs.py
+include: _es_template_yaml
+process_docs: !function utils.process_world_religions
+tag: global_mmlu_full_es_humanities_tasks
+task: global_mmlu_full_es_world_religions

lm_eval/tasks/global_mmlu/full/es/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)