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-13 22:51:54 +00:00
parent a171358f2e
commit 3a85ede8ba
3 changed files with 10 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

4
.secrets.baseline
View file

@ -133,7 +133,7 @@
"filename": "README.md", "filename": "README.md",
"hashed_secret": "a8253456364f1bfc7da7ae4a1db5b45d106317a5", "hashed_secret": "a8253456364f1bfc7da7ae4a1db5b45d106317a5",
"is_verified": false, "is_verified": false,
"line_number": 499 "line_number": 512
} }
], ],
"SLURM.md": [ "SLURM.md": [
@ -561,5 +561,5 @@
} }
] ]
}, },
"generated_at": "2026-03-13T20:13:21Z" "generated_at": "2026-03-13T22:51:44Z"
} }

14
atroposlib/envs/teacher_distillation_env.py
View file

@ -22,17 +22,17 @@ from pydantic import Field
from pydantic_cli import Cmd from pydantic_cli import Cmd
from rich import print as rprint from rich import print as rprint
from .base import BaseEnv, BaseEnvConfig, ScoredDataGroup
from .constants import ENV_NAMESPACE, NAMESPACE_SEP, OPENAI_NAMESPACE
from .server_handling.openai_server import resolve_openai_configs
from .server_handling.server_baseline import APIServerConfig, ServerBaseline
from .server_handling.server_manager import ServerManager, ServerManagerConfig
from ..utils.cli import ( from ..utils.cli import (
extract_namespace, extract_namespace,
get_double_dash_flags, get_double_dash_flags,
get_prefixed_pydantic_model, get_prefixed_pydantic_model,
merge_dicts, merge_dicts,
) )
from .base import BaseEnv, BaseEnvConfig, ScoredDataGroup
from .constants import ENV_NAMESPACE, NAMESPACE_SEP, OPENAI_NAMESPACE
from .server_handling.openai_server import resolve_openai_configs
from .server_handling.server_baseline import APIServerConfig, ServerBaseline
from .server_handling.server_manager import ServerManager, ServerManagerConfig
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@ -81,7 +81,9 @@ class TeacherDistillationEnv(BaseEnv, ABC):
cli_passed_flags: Dict[str, Any], cli_passed_flags: Dict[str, Any],
) -> Optional[Union[ServerBaseline, List[APIServerConfig]]]: ) -> Optional[Union[ServerBaseline, List[APIServerConfig]]]:
teacher_full_prefix = f"{cls.teacher_namespace}{NAMESPACE_SEP}" teacher_full_prefix = f"{cls.teacher_namespace}{NAMESPACE_SEP}"
teacher_cli_passed_args = extract_namespace(cli_passed_flags, teacher_full_prefix) teacher_cli_passed_args = extract_namespace(
cli_passed_flags, teacher_full_prefix
)
yaml_teacher_config = yaml_config.get(cls.teacher_namespace, {}) yaml_teacher_config = yaml_config.get(cls.teacher_namespace, {})
if ( if (

1
atroposlib/tests/test_teacher_distillation_env.py
View file

@ -297,4 +297,3 @@ def test_resolve_teacher_server_configs_uses_teacher_namespace(monkeypatch):
"openai": {"tokenizer_name": "teacher-tokenizer"} "openai": {"tokenizer_name": "teacher-tokenizer"}
} }
assert captured["cli_passed_flags"] == {"openai.base_url": "http://override/v1"} assert captured["cli_passed_flags"] == {"openai.base_url": "http://override/v1"}