Fix utils/notification_service.py (#38556)

* fix

* fix

* update

---------

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2025-06-03 15:59:31 +02:00
committed by GitHub
parent 8cb96787a6
commit e8b292e35f

View File

@@ -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]