mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
c62da98c1a
Upload all run attempts since it can be hard to determine which run attempt to do from HUD, since HUD shows everything together Pull Request resolved: https://github.com/pytorch/pytorch/pull/140459 Approved by: https://github.com/huydhn
28 lines
848 B
Python
28 lines
848 B
Python
import argparse
|
|
import sys
|
|
|
|
from tools.stats.test_dashboard import get_all_run_attempts, upload_additional_info
|
|
from tools.stats.upload_test_stats import get_tests
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser(description="Upload test stats to s3")
|
|
parser.add_argument(
|
|
"--workflow-run-id",
|
|
required=True,
|
|
help="id of the workflow to get artifacts from",
|
|
)
|
|
args = parser.parse_args()
|
|
|
|
print(f"Workflow id is: {args.workflow_run_id}")
|
|
|
|
run_attempts = get_all_run_attempts(args.workflow_run_id)
|
|
|
|
test_cases = []
|
|
for i in run_attempts:
|
|
test_cases = get_tests(args.workflow_run_id, i)
|
|
# Flush stdout so that any errors in the upload show up last in the
|
|
# logs.
|
|
sys.stdout.flush()
|
|
upload_additional_info(args.workflow_run_id, i, test_cases)
|