Skip to content

Commit a26ab95

Browse files
authored
Fix wrong PT/TF categories in CI report (#17272)
Co-authored-by: ydshieh <[email protected]>
1 parent 1ac2b8f commit a26ab95

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

utils/notification_service.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -644,10 +644,10 @@ def add_path(self, path: str, gpu: str = None):
644644
artifact_path["gpu"]
645645
] += f"*{line}*\n_{stacktraces.pop(0)}_\n\n"
646646

647-
if re.search("_tf_", line):
647+
if re.search("test_modeling_tf_", line):
648648
model_results[model]["failed"]["TensorFlow"][artifact_path["gpu"]] += 1
649649

650-
elif re.search("_flax_", line):
650+
elif re.search("test_modeling_flax_", line):
651651
model_results[model]["failed"]["Flax"][artifact_path["gpu"]] += 1
652652

653653
elif re.search("test_modeling", line):

0 commit comments

Comments
 (0)