mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-21 05:34:18 +08:00
747b38c131
See https://github.com/pytorch/pytorch/pull/129751#issue-2380881501. Most changes are auto-generated by linter. You can review these PRs via: ```bash git diff --ignore-all-space --ignore-blank-lines HEAD~1 ``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/129753 Approved by: https://github.com/malfet ghstack dependencies: #129752
40 lines
1.1 KiB
Python
Executable File
40 lines
1.1 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
"""
|
|
Test ownership was introduced in https://github.com/pytorch/pytorch/issues/66232.
|
|
|
|
As a part of enforcing test ownership, we want to maintain a list of existing PyTorch labels
|
|
to verify the owners' existence. This script outputs a file containing a list of existing
|
|
pytorch/pytorch labels so that the file could be uploaded to S3.
|
|
|
|
This script assumes the correct env vars are set for AWS permissions.
|
|
|
|
"""
|
|
|
|
import json
|
|
from typing import Any
|
|
|
|
import boto3 # type: ignore[import]
|
|
from label_utils import gh_get_labels
|
|
|
|
|
|
def parse_args() -> Any:
|
|
from argparse import ArgumentParser
|
|
|
|
parser = ArgumentParser("Export PR labels")
|
|
parser.add_argument("org", type=str)
|
|
parser.add_argument("repo", type=str)
|
|
|
|
return parser.parse_args()
|
|
|
|
|
|
def main() -> None:
|
|
args = parse_args()
|
|
print(f"Exporting labels for {args.org}/{args.repo}")
|
|
labels_file_name = "pytorch_labels.json"
|
|
obj = boto3.resource("s3").Object("ossci-metrics", labels_file_name)
|
|
obj.put(Body=json.dumps(gh_get_labels(args.org, args.repo)).encode())
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|