Tiny update after #34383 (#34404)

* update * update * update --------- Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2024-10-28 12:01:05 +01:00
parent fc465bb196
commit 9360f1827d
3 changed files with 7 additions and 1 deletions
--- a/utils/notification_service.py
+++ b/utils/notification_service.py
@@ -547,7 +547,8 @@ class Message:
                    items = re.findall(pattern, line)
                elif "tests/models/" in line:
                    model = line.split("/")[2]
-                    new_failed_tests[model] = {"single-gpu": [], "multi-gpu": []}
+                    if model not in new_failed_tests:
+                        new_failed_tests[model] = {"single-gpu": [], "multi-gpu": []}
                    for url, device in items:
                        new_failed_tests[model][f"{device}-gpu"].append(line)
            file_path = os.path.join(os.getcwd(), f"ci_results_{job_name}/new_model_failures.json")