diff --git a/applications/ColossalEval/colossal_eval/dataset/__init__.py b/applications/ColossalEval/colossal_eval/dataset/__init__.py index 4e4addd83..02eca8a29 100644 --- a/applications/ColossalEval/colossal_eval/dataset/__init__.py +++ b/applications/ColossalEval/colossal_eval/dataset/__init__.py @@ -3,13 +3,13 @@ from .base import BaseDataset from .ceval import CEvalDataset from .cmmlu import CMMLUDataset from .colossalai import ColossalDataset +from .cvalues import CValuesDataset from .gaokaobench import GaoKaoBenchDataset from .longbench import LongBenchDataset from .mmlu import MMLUDataset from .mtbench import MTBenchDataset from .safetybench_en import SafetyBenchENDataset from .safetybench_zh import SafetyBenchZHDataset -from .cvalues import CValuesDataset __all__ = [ "AGIEvalDataset", @@ -21,7 +21,7 @@ __all__ = [ "MMLUDataset", "ColossalDataset", "MTBenchDataset", - 'SafetyBenchENDataset', - 'SafetyBenchZHDataset', - 'CValuesDataset' + "SafetyBenchENDataset", + "SafetyBenchZHDataset", + "CValuesDataset", ]