From 9b4b807be864e67bffb6ff6108044601d35046db Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Adrian=20W=C3=A4lchli?= <aedu.waelchli@gmail.com>
Date: Wed, 17 Jul 2024 14:52:57 +0000
Subject: [PATCH] set default

---
 src/litdata/utilities/shuffle.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/litdata/utilities/shuffle.py b/src/litdata/utilities/shuffle.py
index 4684838a..cdeb50d5 100644
--- 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]]: