fix data loader hint

Former-commit-id: 49b1e88e3d
This commit is contained in:
hiyouga
2024-06-03 18:28:27 +08:00
parent 72ebcb9a04
commit 64d24842fe

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: