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

[pre-commit.ci] auto fixes from pre-commit.com hooks

Browse source 
for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2026-03-04 03:08:15 +00:00
parent 5aaf7a346c
commit 8f304d44fd
1 changed files with 2 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

8
atroposlib/envs/server_handling/managed_server.py
View file

@ -581,9 +581,7 @@ class ManagedServer:
required = ("prompt_tokens", "prompt_topk_token_ids", "prompt_topk_logprobs") required = ("prompt_tokens", "prompt_topk_token_ids", "prompt_topk_logprobs")
missing = [k for k in required if k not in payload] missing = [k for k in required if k not in payload]
if missing: if missing:
raise ValueError( raise ValueError(f"get_logprobs response missing required keys: {missing}")
f"get_logprobs response missing required keys: {missing}"
)
prompt_tokens = payload["prompt_tokens"] prompt_tokens = payload["prompt_tokens"]
token_ids = payload["prompt_topk_token_ids"] token_ids = payload["prompt_topk_token_ids"]
@ -596,9 +594,7 @@ class ManagedServer:
"prompt_topk_token_ids and prompt_topk_logprobs must be list-of-list." "prompt_topk_token_ids and prompt_topk_logprobs must be list-of-list."
) )
if len(token_ids) != len(prompt_tokens) or len(logprobs) != len(prompt_tokens): if len(token_ids) != len(prompt_tokens) or len(logprobs) != len(prompt_tokens):
raise ValueError( raise ValueError("prompt_topk arrays must align with prompt_tokens length.")
"prompt_topk arrays must align with prompt_tokens length."
)
for idx, (tok_row, lp_row) in enumerate(zip(token_ids, logprobs)): for idx, (tok_row, lp_row) in enumerate(zip(token_ids, logprobs)):
if not isinstance(tok_row, list) or not isinstance(lp_row, list): if not isinstance(tok_row, list) or not isinstance(lp_row, list):