diff --git a/src/litdata/utilities/dataset_utilities.py b/src/litdata/utilities/dataset_utilities.py index f7d39bc6..0105c0a0 100644 --- a/src/litdata/utilities/dataset_utilities.py +++ b/src/litdata/utilities/dataset_utilities.py @@ -116,7 +116,7 @@ def generate_roi(chunks: List[Dict[str, Any]], item_loader: Optional[BaseItemLoa return roi -def load_index_file(input_dir: str) -> Dict[str, Any] or None: +def load_index_file(input_dir: str) -> Dict[str, Any]: """Load index file from the input_dir.""" index_filepath = os.path.join(input_dir, _INDEX_FILENAME)