set default

Lightning-AI · Jul 17, 2024 · 9b4b807 · 9b4b807
1 parent 0641666
commit 9b4b807
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/litdata/utilities/shuffle.py b/src/litdata/utilities/shuffle.py
@@ -47,7 +47,7 @@ def _associate_chunks_and_internals_to_workers(
     distributed_env: _DistributedEnv,
     indexes: Any,
     chunk_intervals: List[Interval],
-    drop_last: bool,
+    drop_last: bool = False,
     num_workers: int = 1,
     batch_size: int = 1,
 ) -> Tuple[List[List[int]], List[Any]]: