open-thought/reasoning-gym
1
0
Fork 0
mirror of https://github.com/open-thought/reasoning-gym.git synced 2026-04-28 17:29:39 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: Move EpochTrackingDataLoader after ReasoningGymDataset to resolve undefined name error

Browse source
This commit is contained in:
Andreas Koepf (aider) 2025-02-22 21:12:15 +00:00 • committed by Andreas Koepf
parent 5f16d54ebe
commit 8dc6cb5228
4 changed files with 24 additions and 30 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tools/server/server.py
View file

@ -76,12 +76,12 @@ def create_app(config: ServerConfig) -> FastAPI:
def permute_index(idx: int, epoch_seed: int, dataset_size: int) -> int:
"""Generate a deterministic permuted index without materializing full permutation.
Args:
idx: Original index to permute
epoch_seed: Seed for this epoch's permutation
dataset_size: Size of the dataset
Returns:
Permuted index in range [0, dataset_size)
"""
@ -107,7 +107,7 @@ def create_app(config: ServerConfig) -> FastAPI:
dataset_size = len(experiment.dataset)
base_seed = experiment.config.seed if experiment.config.seed is not None else 0
epoch_seed = base_seed + (epoch * dataset_size)
entries = []
for i in range(base_index, base_index + batch_size):
# Get permuted index for this position