From e2d3f4b4e64aaa83f54ee3b2e5c6b759850615a6 Mon Sep 17 00:00:00 2001 From: "Andreas Koepf (aider)" Date: Thu, 30 Jan 2025 22:05:14 +0100 Subject: [PATCH] feat: Add seed wrapping at 2^32 to prevent unbounded growth --- reasoning_gym/dataset.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/reasoning_gym/dataset.py b/reasoning_gym/dataset.py index 4e271a50..5825765b 100644 --- a/reasoning_gym/dataset.py +++ b/reasoning_gym/dataset.py @@ -93,8 +93,8 @@ class ReseedingDataset(Iterable[Dict[str, Any]]): # Create new config with modified seed new_config = deepcopy(self.dataset.config) if hasattr(new_config, "seed"): - # Derive new seed from chunk number using dataset's seed - new_config.seed = self.dataset.seed + chunk_num + # Derive new seed from chunk number using dataset's seed, wrapping around at 2^32 + new_config.seed = (self.dataset.seed + chunk_num) % (2**32) # Create new dataset instance with chunk config return self.dataset_cls(new_config)