fix data loader hint

This commit is contained in:
hiyouga 2024-06-03 18:28:27 +08:00
parent c4f50865ad
commit 49b1e88e3d
1 changed files with 2 additions and 2 deletions

View File

@ -62,9 +62,9 @@ def load_single_dataset(
raise ValueError("File {} not found.".format(local_path))
if data_path is None:
raise ValueError("File extension must be txt, csv, json or jsonl.")
raise ValueError("Allowed file types: {}.".format(",".join(FILEEXT2TYPE.keys())))
else:
raise NotImplementedError
raise NotImplementedError("Unknown load type: {}.".format(dataset_attr.load_from))
if dataset_attr.load_from == "ms_hub":
try: