mirror of
https://github.com/open-thought/reasoning-gym.git
synced 2025-10-09 13:40:09 +03:00
Run categories in parallel (#492)
This commit is contained in:
@@ -659,11 +659,11 @@ class AsyncModelEvaluator:
|
||||
|
||||
# Process each category sequentially to ensure proper checkpointing
|
||||
category_results = []
|
||||
for category in self.config.categories:
|
||||
category_result = await self.evaluate_category(category)
|
||||
category_results.append(category_result)
|
||||
tasks = [asyncio.create_task(self.evaluate_category(category)) for category in self.config.categories]
|
||||
|
||||
# Update partial summary after each category
|
||||
for finished_task in asyncio.as_completed(tasks):
|
||||
result = await finished_task
|
||||
category_results.append(result)
|
||||
self._update_partial_summary(category_results)
|
||||
|
||||
# Generate results structure
|
||||
|
||||
Reference in New Issue
Block a user