From a26ab95e302a26cf44c0cf8a5717a3680bf2eb96 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Tue, 17 May 2022 09:32:47 +0200 Subject: [PATCH] Fix wrong PT/TF categories in CI report (#17272) Co-authored-by: ydshieh --- utils/notification_service.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/utils/notification_service.py b/utils/notification_service.py index 0b7e8387fb..9323079d55 100644 --- a/utils/notification_service.py +++ b/utils/notification_service.py @@ -644,10 +644,10 @@ if __name__ == "__main__": artifact_path["gpu"] ] += f"*{line}*\n_{stacktraces.pop(0)}_\n\n" - if re.search("_tf_", line): + if re.search("test_modeling_tf_", line): model_results[model]["failed"]["TensorFlow"][artifact_path["gpu"]] += 1 - elif re.search("_flax_", line): + elif re.search("test_modeling_flax_", line): model_results[model]["failed"]["Flax"][artifact_path["gpu"]] += 1 elif re.search("test_modeling", line):