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

for more information, see https://pre-commit.ci

.secrets.baseline

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

atroposlib/envs/teacher_distillation_env.py

@@ -22,17 +22,17 @@ from pydantic import Field
 from pydantic_cli import Cmd
 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 (
     extract_namespace,
     get_double_dash_flags,
     get_prefixed_pydantic_model,
     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__)
 
@@ -81,7 +81,9 @@ class TeacherDistillationEnv(BaseEnv, ABC):
         cli_passed_flags: Dict[str, Any],
     ) -> Optional[Union[ServerBaseline, List[APIServerConfig]]]:
         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, {})
 
         if (

atroposlib/tests/test_teacher_distillation_env.py

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