file->folder

mosaicml · Sep 19, 2024 · f026bff · f026bff
1 parent a439f01
commit f026bff
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/llmfoundry/command_utils/data_prep/convert_text_to_mds.py b/llmfoundry/command_utils/data_prep/convert_text_to_mds.py
@@ -31,8 +31,8 @@
 from llmfoundry.utils.exceptions import (
     CannotUnicodeDecodeFile,
     DatasetTooSmallError,
-    InputFileNotFound,
     InputFolderMissingDataError,
+    InputFolderNotFound,
     OutputFolderNotEmptyError,
 )
 
@@ -133,7 +133,7 @@ def get_object_names(input_folder: str) -> list[str]:
             ]
             log.info(f'Found {len(names)} text files in remote storage')
         except FileNotFoundError:
-            raise InputFileNotFound(folder_prefix)
+            raise InputFolderNotFound(folder_prefix)
 
     else:
         # input_folder is a local folder