From e8b292e35f331d3c3de85f7e5d3496b0e13d3d6f Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Tue, 3 Jun 2025 15:59:31 +0200 Subject: [PATCH] Fix `utils/notification_service.py` (#38556) * fix * fix * update --------- Co-authored-by: ydshieh --- utils/notification_service.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/utils/notification_service.py b/utils/notification_service.py index 43f132bf12d..b8e997ccc71 100644 --- a/utils/notification_service.py +++ b/utils/notification_service.py @@ -615,7 +615,10 @@ class Message: pattern = r"<(https://github.com/huggingface/transformers/actions/runs/.+?/job/.+?)\|(.+?)>" items = re.findall(pattern, line) elif "tests/" in line: - if "tests/models/" in line or "tests/quantization/" in line: + # TODO: Improve the condition here. + if "tests/models/" in line or ( + "tests/quantization/" in line and job_name == "run_quantization_torch_gpu" + ): model = line.split("/")[2] else: model = line.split("/")[1]