mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
8a67daf283
Pull Request resolved: https://github.com/pytorch/pytorch/pull/129375 Approved by: https://github.com/malfet
31 lines
907 B
Python
31 lines
907 B
Python
import argparse
|
|
import sys
|
|
|
|
from tools.stats.test_dashboard import upload_additional_info
|
|
from tools.stats.upload_test_stats import get_tests
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser(description="Upload test stats to Rockset")
|
|
parser.add_argument(
|
|
"--workflow-run-id",
|
|
required=True,
|
|
help="id of the workflow to get artifacts from",
|
|
)
|
|
parser.add_argument(
|
|
"--workflow-run-attempt",
|
|
type=int,
|
|
required=True,
|
|
help="which retry of the workflow this is",
|
|
)
|
|
args = parser.parse_args()
|
|
|
|
print(f"Workflow id is: {args.workflow_run_id}")
|
|
|
|
test_cases = get_tests(args.workflow_run_id, args.workflow_run_attempt)
|
|
|
|
# Flush stdout so that any errors in Rockset upload show up last in the logs.
|
|
sys.stdout.flush()
|
|
|
|
upload_additional_info(args.workflow_run_id, args.workflow_run_attempt, test_cases)
|