NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-19 12:57:58 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update server.py

Browse source
This commit is contained in:
Nina 2025-11-07 19:03:54 +01:00 • committed by GitHub
parent 74b5412c2b
commit b3feae5eef
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 9 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

23
atroposlib/api/server.py
View file

@ -416,20 +416,15 @@ async def scored_data_list(scored_data_list: List[ScoredData]):
buffered_count = 0
last_buffer_size: Optional[int] = None
for scored_data in scored_data_list:
data_dict = {
"tokens": scored_data.tokens,
"masks": scored_data.masks,
"scores": scored_data.scores,
"advantages": scored_data.advantages,
"ref_logprobs": scored_data.ref_logprobs,
"images": scored_data.images,
"messages": scored_data.messages,
"generation_params": scored_data.generation_params,
"inference_logprobs": scored_data.inference_logprobs,
"overrides": scored_data.overrides,
"group_overrides": scored_data.group_overrides,
"env_id": scored_data.env_id,
}
result = _process_scored_data(scored_data)
if result.get("status") == "buffered":
buffered_count += 1
last_buffer_size = result.get("buffer_size", last_buffer_size)
response: Dict[str, Any] = {
"status": "received",
"groups_processed": len(scored_data_list),
}
# Check if this is a mixed-size group
env_id = scored_data.env_id