diff --git a/.secrets.baseline b/.secrets.baseline
index 2651783a..31d5870f 100644
--- a/.secrets.baseline
+++ b/.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"
 }
diff --git a/atroposlib/envs/teacher_distillation_env.py b/atroposlib/envs/teacher_distillation_env.py
index 1b3cda8f..892d424c 100644
--- a/atroposlib/envs/teacher_distillation_env.py
+++ b/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 (
diff --git a/atroposlib/tests/test_teacher_distillation_env.py b/atroposlib/tests/test_teacher_distillation_env.py
index c789670d..b348e65a 100644
--- a/atroposlib/tests/test_teacher_distillation_env.py
+++ b/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"}
-