From e33a90552df4532b1bc0c2de39d1aa8c829a8331 Mon Sep 17 00:00:00 2001 From: David Hall Date: Fri, 13 Sep 2024 00:48:59 -0700 Subject: [PATCH] ugh --- src/levanter/data/text.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/levanter/data/text.py b/src/levanter/data/text.py index ad03f6d01..7e25c88ef 100644 --- a/src/levanter/data/text.py +++ b/src/levanter/data/text.py @@ -600,7 +600,7 @@ def train_set( if self.shuffle is True: ds = ds.shuffle(key) - elif isinstance(self.shuffle, int): + elif isinstance(self.shuffle, int) and self.shuffle > 0: ds = ds.era_shuffle(self.shuffle, key=key) return ds # type: ignore