From 42f6a742c9c3984b29ccf6e74be78c2d3f3411dc Mon Sep 17 00:00:00 2001 From: "Andreas Koepf (aider)" Date: Thu, 30 Jan 2025 22:02:21 +0100 Subject: [PATCH] refactor: Use self.dataset.seed directly for chunk seed generation --- reasoning_gym/dataset.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/reasoning_gym/dataset.py b/reasoning_gym/dataset.py index 6a979290..4e271a50 100644 --- a/reasoning_gym/dataset.py +++ b/reasoning_gym/dataset.py @@ -93,9 +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 - base_seed = new_config.seed if new_config.seed is not None else 0 - new_config.seed = base_seed + chunk_num + # Derive new seed from chunk number using dataset's seed + new_config.seed = self.dataset.seed + chunk_num # Create new dataset instance with chunk config return self.dataset_cls(new_config)