update

huggingface · Mar 28, 2024 · a2b9845 · a2b9845
1 parent d636af2
commit a2b9845
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/utils/notification_service.py b/utils/notification_service.py
@@ -1072,7 +1072,8 @@ def prepare_reports(title, header, reports, to_truncate=True):
         additional_files = {}
 
     job_name = os.getenv("CI_TEST_JOB")
-    test_name = job_to_test_map[job_name]
+    if job_name in job_to_test_map:
+        test_name = job_to_test_map[job_name]
 
     additional_files = {k: v for k, v in additional_files.items() if k == test_name}