diff --git a/.secrets.baseline b/.secrets.baseline index 7059d28b..e4785ede 100644 --- a/.secrets.baseline +++ b/.secrets.baseline @@ -133,7 +133,7 @@ "filename": "README.md", "hashed_secret": "a8253456364f1bfc7da7ae4a1db5b45d106317a5", "is_verified": false, - "line_number": 454 + "line_number": 495 } ], "SLURM.md": [ @@ -561,5 +561,5 @@ } ] }, - "generated_at": "2026-03-02T22:46:56Z" + "generated_at": "2026-03-13T17:20:46Z" } diff --git a/atroposlib/envs/teacher_distillation_env.py b/atroposlib/envs/teacher_distillation_env.py index d8284335..85e040c7 100644 --- a/atroposlib/envs/teacher_distillation_env.py +++ b/atroposlib/envs/teacher_distillation_env.py @@ -89,7 +89,9 @@ class TeacherDistillationEnv(BaseEnv, ABC): # Core fetch # ------------------------------------------------------------------ - def _validate_teacher_tokenizer_compatibility(self, teacher_tokenizer_name: str) -> None: + def _validate_teacher_tokenizer_compatibility( + self, teacher_tokenizer_name: str + ) -> None: student_tok_name = getattr(self.tokenizer, "name_or_path", None) or "" if student_tok_name == teacher_tokenizer_name: return diff --git a/atroposlib/tests/test_teacher_distillation_env.py b/atroposlib/tests/test_teacher_distillation_env.py index 7f5262e7..65262984 100644 --- a/atroposlib/tests/test_teacher_distillation_env.py +++ b/atroposlib/tests/test_teacher_distillation_env.py @@ -88,7 +88,9 @@ def test_teacher_tokenizer_mismatch_raises(monkeypatch): lambda *args, **kwargs: _TeacherTokenizer(), ) - with pytest.raises(ValueError, match="Cross-tokenizer distillation is not supported"): + with pytest.raises( + ValueError, match="Cross-tokenizer distillation is not supported" + ): TeacherDistillationEnv._validate_teacher_tokenizer_compatibility( env, teacher_tokenizer_name="teacher-model",