From c8c88b943adaa7700089ef7a024232e3fde71df4 Mon Sep 17 00:00:00 2001 From: bhimrazy Date: Sat, 6 Jul 2024 16:34:07 +0545 Subject: [PATCH] fix: return type --- src/litdata/utilities/dataset_utilities.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)