diff --git a/utils/notification_service.py b/utils/notification_service.py index a525a1eded..25933a0ab2 100644 --- a/utils/notification_service.py +++ b/utils/notification_service.py @@ -819,8 +819,8 @@ if __name__ == "__main__": stacktraces = handle_stacktraces(artifact["failures_line"]) for line in artifact["summary_short"].split("\n"): - if re.search("FAILED", line): - line = line.replace("FAILED ", "") + if line.startswith("FAILED "): + line = line[len("FAILED ") :] line = line.split()[0].replace("\n", "") if artifact_path["gpu"] not in model_results[model]["failures"]: @@ -910,8 +910,8 @@ if __name__ == "__main__": if failed: for line in artifact["summary_short"].split("\n"): - if re.search("FAILED", line): - line = line.replace("FAILED ", "") + if line.startswith("FAILED "): + line = line[len("FAILED ") :] line = line.split()[0].replace("\n", "") if artifact_path["gpu"] not in additional_results[key]["failures"]: