Update mtbench101.py (#1276)

fix wrong-used import
from torch.utils.data import DataLoader, Dataset
This commit is contained in:
Xingyuan Bu 2024-06-26 00:40:22 +08:00 committed by GitHub
parent 26d077b080
commit 842fb1cd70
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -5,7 +5,6 @@ import re
from typing import Optional
from datasets import Dataset, DatasetDict
from torch.utils.data import DataLoader, Dataset
from opencompass.registry import LOAD_DATASET