diff --git a/opencompass/datasets/rbench.py b/opencompass/datasets/rbench.py index a2f03950..986cdaeb 100644 --- a/opencompass/datasets/rbench.py +++ b/opencompass/datasets/rbench.py @@ -1,5 +1,7 @@ from datasets import Dataset, load_dataset + from opencompass.registry import LOAD_DATASET + from .base import BaseDataset