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

refactor: Add optional epoch parameter to generate_batch endpoint

Browse source
This commit is contained in:
Andreas Koepf 2025-02-22 20:24:33 +00:00 • committed by Andreas Koepf (aider)
parent e4102a44f6
commit 1864a54d53
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
tools/server/server.py
View file

@ -74,7 +74,7 @@ def create_app(config: ServerConfig) -> FastAPI:
return {"status": "deleted"}
@app.get("/experiments/{name}/batch", response_model=BatchResponse)
async def generate_batch(name: str, base_index: int, batch_size: int):
async def generate_batch(name: str, base_index: int, batch_size: int, epoch: int = 0):
"""Generate a batch of raw entries"""
# Validate parameters
if base_index < 0: