riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-03 04:40:06 +06:00
Code Issues Actions 28 Packages Projects Releases Wiki Activity

Fix utils/notification_service.py (#38556)

Browse Source 
* 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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
utils/notification_service.py
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]