fix value error

mosaicml · Dec 19, 2023 · 2e4fb93 · 2e4fb93
1 parent 0dc2049
commit 2e4fb93
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/llmfoundry/data/finetuning/tasks.py b/llmfoundry/data/finetuning/tasks.py
@@ -421,7 +421,7 @@ def build_from_hf(
                         for f in dataset_files):
                     raise ValueError(
                         f'Dataset at local path {dataset_name} contains invalid file types. '
-                        + 'Allowed file types are: .csv, .jsonl, and .parquet.')
+                        + f'Allowed file types are: {SUPPORTED_EXTENSIONS}')
             dataset = hf_datasets.load_dataset(dataset_name,
                                                split=split,
                                                **hf_kwargs)