diff --git a/opencompass/configs/datasets/SeedBench/seedbench_gen_5d5ea1.py b/opencompass/configs/datasets/SeedBench/seedbench_gen_5d5ea1.py index 55babe53..8132f4d6 100644 --- a/opencompass/configs/datasets/SeedBench/seedbench_gen_5d5ea1.py +++ b/opencompass/configs/datasets/SeedBench/seedbench_gen_5d5ea1.py @@ -2,7 +2,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.openicl.icl_evaluator import AccEvaluator, JiebaRougeEvaluator, RougeEvaluator -from opencompass.datasets.SeedBench import SeedBenchDataset, F1ScoreEvaluator, my_multiple_select_postprocess, AverageRougeScoreEvaluator +from opencompass.datasets import SeedBenchDataset, F1ScoreEvaluator, my_multiple_select_postprocess, AverageRougeScoreEvaluator from opencompass.utils.text_postprocessors import first_option_postprocess diff --git a/opencompass/datasets/SeedBench.py b/opencompass/datasets/SeedBench.py index 7b88ea05..2f4e0b33 100644 --- a/opencompass/datasets/SeedBench.py +++ b/opencompass/datasets/SeedBench.py @@ -26,7 +26,7 @@ class SeedBenchDataset(BaseDataset): **kwargs) -> datasets.Dataset: path = get_data_path(path) - if environ.get('DATASET_SOURCE', None) == 'ModelScope': + if environ.get('DATASET_SOURCE') == 'ModelScope': from modelscope import MsDataset dataset = MsDataset.load(path, subset_name='default',