NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-27 17:23:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

[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] 2025-10-16 17:39:00 +00:00
parent c36ec29656
commit 1e6a745491
3 changed files with 59 additions and 34 deletions
Download patch file Download diff file Expand all files Collapse all files

6
atroposlib/envs/server_handling/server_manager.py
View file

@ -15,8 +15,8 @@ from atroposlib.envs.server_handling.server_baseline import (
ServerBaseline,
)
from atroposlib.envs.server_handling.server_harness import ServerHarness
from atroposlib.envs.server_handling.trl_vllm_server import TrlVllmServer
from atroposlib.envs.server_handling.sglang_server import SGLangServer
from atroposlib.envs.server_handling.trl_vllm_server import TrlVllmServer
class ServerManagerConfig(BaseModel):
@ -289,7 +289,9 @@ class ServerManager:
most_available_server_num_slots = (
server.sem._value if is_train else server.eval_sem._value
)
return await self.servers[most_available_server].tokens_and_logprobs_completion(**kwargs)
return await self.servers[most_available_server].tokens_and_logprobs_completion(
**kwargs
)
@asynccontextmanager
async def dedicated_server(self) -> AsyncGenerator[OpenAIServer, None]: