Skip to content

GitLab

Unverified Commit a26ab95e authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Fix wrong PT/TF categories in CI report (#17272) 


Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent 1ac2b8fa
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
utils/notification_service.py
View file @ a26ab95e
......@@ -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):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment