diff --git a/opencompass/configs/datasets/aime2024/aime2024_llmjudge_gen_5e9f4f.py b/opencompass/configs/datasets/aime2024/aime2024_llmjudge_gen_5e9f4f.py index c1be29dd..a54e53f1 100644 --- a/opencompass/configs/datasets/aime2024/aime2024_llmjudge_gen_5e9f4f.py +++ b/opencompass/configs/datasets/aime2024/aime2024_llmjudge_gen_5e9f4f.py @@ -1,7 +1,7 @@ from opencompass.openicl.icl_prompt_template import PromptTemplate from opencompass.openicl.icl_retriever import ZeroRetriever from opencompass.openicl.icl_inferencer import GenInferencer -from opencompass.datasets import CustomDataset +from opencompass.datasets import Aime2024Dataset from opencompass.evaluator import GenericLLMEvaluator from opencompass.datasets import generic_llmjudge_postprocess @@ -69,7 +69,7 @@ aime2024_eval_cfg = dict( ), ), dataset_cfg=dict( - type=CustomDataset, + type=Aime2024Dataset, path='opencompass/aime2024', reader_cfg=aime2024_reader_cfg, ), @@ -81,7 +81,7 @@ aime2024_eval_cfg = dict( aime2024_datasets = [ dict( abbr='aime2024', - type=CustomDataset, + type=Aime2024Dataset, path='opencompass/aime2024', reader_cfg=aime2024_reader_cfg, infer_cfg=aime2024_infer_cfg,