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)