File size: 2,555 Bytes
4f4b027 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 | # /// script
# requires-python = ">=3.10"
# dependencies = [
# "requests>=2.31.0",
# ]
# ///
"""
Check for open pull requests on a Hugging Face model repository.
Usage:
uv run scripts/check_prs.py --repo-id "org/model-name"
"""
import argparse
import sys
from typing import Any
import requests
def get_open_prs(repo_id: str) -> list[dict[str, Any]]:
"""
Fetch open pull requests for a Hugging Face model repository.
Args:
repo_id: Hugging Face model repository ID (e.g., "nvidia/model-name")
Returns:
List of open PR dictionaries with num, title, author, and createdAt
"""
url = f"https://huggingface.co/api/models/{repo_id}/discussions"
try:
response = requests.get(url, timeout=30, allow_redirects=True)
response.raise_for_status()
data = response.json()
discussions = data.get("discussions", [])
open_prs = [
{
"num": d["num"],
"title": d["title"],
"author": d["author"]["name"],
"createdAt": d.get("createdAt", "unknown"),
}
for d in discussions
if d.get("status") == "open" and d.get("isPullRequest")
]
return open_prs
except requests.RequestException as e:
print(f"Error fetching PRs from Hugging Face: {e}", file=sys.stderr)
return []
def list_open_prs(repo_id: str) -> None:
"""Display open pull requests for a model repository."""
prs = get_open_prs(repo_id)
print(f"\n{'='*70}")
print(f"Open Pull Requests for: {repo_id}")
print(f"{'='*70}")
if not prs:
print("\nNo open pull requests found.")
else:
print(f"\nFound {len(prs)} open PR(s):\n")
for pr in prs:
print(f" PR #{pr['num']} - {pr['title']}")
print(f" Author: {pr['author']}")
print(f" Created: {pr['createdAt']}")
print(f" URL: https://huggingface.co/{repo_id}/discussions/{pr['num']}")
print()
print(f"{'='*70}\n")
def main():
parser = argparse.ArgumentParser(
description="Check for open pull requests on a Hugging Face model repository.",
epilog="Always run this before creating new PRs to avoid duplicates.",
)
parser.add_argument(
"--repo-id",
type=str,
required=True,
help="HF repository ID (e.g., 'nvidia/model-name')",
)
args = parser.parse_args()
list_open_prs(args.repo_id)
if __name__ == "__main__":
main()
|