diff --git a/opencompass/datasets/rbench.py b/opencompass/datasets/rbench.py index ddbddc6b..a2f03950 100644 --- a/opencompass/datasets/rbench.py +++ b/opencompass/datasets/rbench.py @@ -1,4 +1,4 @@ -from datasets import Dataset, DatasetDict, load_dataset +from datasets import Dataset, load_dataset from opencompass.registry import LOAD_DATASET from .base import BaseDataset