From 4130f50e07151329c3fe2186db10a1f90a3b2b89 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Tue, 16 Jul 2024 22:27:42 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- src/litdata/streaming/dataset.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/litdata/streaming/dataset.py b/src/litdata/streaming/dataset.py index 090adba6..62c4228c 100644 --- a/src/litdata/streaming/dataset.py +++ b/src/litdata/streaming/dataset.py @@ -473,6 +473,7 @@ def is_integer(value: str) -> bool: except Exception: return False + def _replay_sampling(num_samples_yielded: int, batch_size: int, num_workers: int) -> Dict[int, int]: """This function replays the sampling from the dataloader.""" divisible_num_batches_yielded = num_samples_yielded // (num_workers * batch_size)