We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 76cc1e7 commit 2d97758Copy full SHA for 2d97758
eval_protocol/pytest/remote_rollout_processor.py
@@ -169,9 +169,14 @@ def _load_data():
169
else:
170
raise ValueError("RemoteRolloutProcessor's output_data_loader should return exactly one row.")
171
172
- for r in rows:
173
- tasks.append(asyncio.create_task(_process_row(r)))
+ semaphore = config.semaphore
174
+ async def _sem_wrapper(r: EvaluationRow) -> EvaluationRow:
175
+ async with semaphore:
176
+ result = await _process_row(r)
177
+ return result
178
+
179
+ tasks = [asyncio.create_task(_sem_wrapper(row)) for row in rows]
180
return tasks
181
182
def cleanup(self) -> None:
0 commit comments