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

clean logging

Browse source
This commit is contained in:
Jai Suphavadeeprasit 2026-03-13 12:38:52 -04:00
parent 600c54f5f8
commit 862cd3667d
3 changed files with 24 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -1,6 +1,5 @@
import asyncio
import inspect
import logging
import os
import warnings
from contextlib import asynccontextmanager
@ -26,8 +25,6 @@ from atroposlib.envs.server_handling.sglang_server import SGLangServer
from atroposlib.envs.server_handling.trl_vllm_server import TrlVllmServer
from atroposlib.envs.server_handling.vllm_server import VLLMServer
logger = logging.getLogger(__name__)
class ServerManagerConfig(BaseModel):
slurm: bool = Field(

3
atroposlib/envs/server_handling/vllm_server.py
View file

@ -2,7 +2,6 @@
# see example_trainer/vllm_api_server.py for an example
import asyncio
import logging
import warnings
from typing import Any, Dict, List, Tuple
@ -20,8 +19,6 @@ from atroposlib.envs.server_handling.server_baseline import (
ReasoningConfig,
)
logger = logging.getLogger(__name__)
class VLLMServer(APIServer):
"""