mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-03 12:50:06 +06:00
f2e197c30a
* First draft of github action on PR opening for auto-assigning reviewers * fix missing import * Don't reassign reviewers if we already have them * Temporarily comment out the opened line so we can test the script * Correct path for codeowners file * Update workflow permissions * Update workflow permissions * Update debug logs * Strip inline comments * Remove prefix * Request reviews instead of assigning * Request reviews instead of assigning * Add TODO * Use pull-request-target instead * Update the script * Set back to pull_request for testing * Set to pull_request_target, testing works! * Add licence * Tighten up one of the globs * Refactor things to be a bit less convoluted * Only assign reviewers when marked ready for review
88 lines
3.0 KiB
Python
88 lines
3.0 KiB
Python
# coding=utf-8
|
|
# Copyright 2025 the HuggingFace Inc. team. All rights reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
import os
|
|
import github
|
|
import json
|
|
from github import Github
|
|
from fnmatch import fnmatch
|
|
from collections import Counter
|
|
from pathlib import Path
|
|
|
|
def get_file_owners(file_path, codeowners_lines):
|
|
# Process lines in reverse (last matching pattern takes precedence)
|
|
for line in reversed(codeowners_lines):
|
|
# Skip comments and empty lines, strip inline comments
|
|
line = line.split('#')[0].strip()
|
|
if not line:
|
|
continue
|
|
|
|
# Split into pattern and owners
|
|
parts = line.split()
|
|
pattern = parts[0]
|
|
# Can be empty, e.g. for dummy files with explicitly no owner!
|
|
owners = [owner.removeprefix("@") for owner in parts[1:]]
|
|
|
|
# Check if file matches pattern
|
|
if fnmatch(file_path, pattern):
|
|
return owners # Remember, can still be empty!
|
|
return [] # Should never happen, but just in case
|
|
|
|
def main():
|
|
g = Github(os.environ['GITHUB_TOKEN'])
|
|
repo = g.get_repo("huggingface/transformers")
|
|
with open(os.environ['GITHUB_EVENT_PATH']) as f:
|
|
event = json.load(f)
|
|
script_dir = Path(__file__).parent.absolute()
|
|
with open(script_dir / "codeowners_for_review_action") as f:
|
|
codeowners_lines = f.readlines()
|
|
|
|
# The PR number is available in the event payload
|
|
pr_number = event['pull_request']['number']
|
|
pr = repo.get_pull(pr_number)
|
|
pr_author = pr.user.login
|
|
|
|
existing_reviews = list(pr.get_reviews())
|
|
if existing_reviews:
|
|
print(f"Already has reviews: {[r.user.login for r in existing_reviews]}")
|
|
return
|
|
|
|
users_requested, teams_requested = pr.get_review_requests()
|
|
users_requested = list(users_requested)
|
|
if users_requested:
|
|
print(f"Reviewers already requested: {users_requested}")
|
|
return
|
|
|
|
locs_per_owner = Counter()
|
|
for file in pr.get_files():
|
|
owners = get_file_owners(file.filename, codeowners_lines)
|
|
for owner in owners:
|
|
locs_per_owner[owner] += file.changes
|
|
|
|
# Assign the top 2 based on locs changed as reviewers, but skip the owner if present
|
|
locs_per_owner.pop(pr_author, None)
|
|
top_owners = locs_per_owner.most_common(2)
|
|
print("Top owners", top_owners)
|
|
top_owners = [owner[0] for owner in top_owners]
|
|
try:
|
|
pr.create_review_request(top_owners)
|
|
except github.GithubException as e:
|
|
print(f"Failed to request review for {top_owners}: {e}")
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|