diff --git a/.secrets.baseline b/.secrets.baseline
index 31d5870f..f93f324a 100644
--- a/.secrets.baseline
+++ b/.secrets.baseline
@@ -133,7 +133,7 @@
         "filename": "README.md",
         "hashed_secret": "a8253456364f1bfc7da7ae4a1db5b45d106317a5",
         "is_verified": false,
-        "line_number": 512
+        "line_number": 530
       }
     ],
     "SLURM.md": [
@@ -561,5 +561,5 @@
       }
     ]
   },
-  "generated_at": "2026-03-13T22:51:44Z"
+  "generated_at": "2026-03-14T00:43:09Z"
 }
diff --git a/atroposlib/tests/test_teacher_distillation_env.py b/atroposlib/tests/test_teacher_distillation_env.py
index b35c48ee..11c586d9 100644
--- a/atroposlib/tests/test_teacher_distillation_env.py
+++ b/atroposlib/tests/test_teacher_distillation_env.py
@@ -205,7 +205,9 @@ def test_init_requires_teacher_server_source(monkeypatch):
         teacher_enabled=True,
         teacher_top_k=0,
     )
-    with pytest.raises(ValueError, match="no teacher server configuration was provided"):
+    with pytest.raises(
+        ValueError, match="no teacher server configuration was provided"
+    ):
         _ConcreteTeacherEnv(
             config=config,
             server_configs=[],