ultralytics-actions 0.0.25__tar.gz → 0.0.26__tar.gz

{ultralytics_actions-0.0.25 → ultralytics_actions-0.0.26}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ultralytics-actions
-Version: 0.0.25
+Version: 0.0.26
 Summary: Ultralytics Actions for GitHub automation and PR management.
 Author-email: Glenn Jocher <glenn.jocher@ultralytics.com>
 Maintainer-email: Ultralytics <hello@ultralytics.com>

{ultralytics_actions-0.0.25 → ultralytics_actions-0.0.26}/actions/__init__.py

@@ -22,4 +22,4 @@
 #     ├── test_summarize_pr.py
 #     └── ...
 
-__version__ = "0.0.25"
+__version__ = "0.0.26"

{ultralytics_actions-0.0.25 → ultralytics_actions-0.0.26}/actions/summarize_pr.py

@@ -13,7 +13,7 @@ from .utils import (
     get_pr_diff,
 )
 
-# Action settings
+# Constants
 SUMMARY_START = (
     "## 🛠️ PR Summary\n\n<sub>Made with ❤️ by [Ultralytics Actions](https://github.com/ultralytics/actions)<sub>\n\n"
 )
@@ -111,12 +111,20 @@ query($owner: String!, $repo: String!, $pr_number: Int!) {
         return
 
 
+def remove_todos_on_merge(pr_number):
+    """Removes specified labels from PR."""
+    for label in ["TODO"]:  # Can be extended with more labels in the future
+        requests.delete(
+            f"{GITHUB_API_URL}/repos/{GITHUB_REPOSITORY}/issues/{pr_number}/labels/{label}", headers=GITHUB_HEADERS
+        )
+
+
 def main():
     """Summarize a pull request and update its description with an AI-generated summary."""
     pr_number = PR["number"]
 
     print(f"Retrieving diff for PR {pr_number}")
-    diff = get_pr_diff(PR["number"])
+    diff = get_pr_diff(pr_number)
 
     # Generate PR summary
     print("Generating PR summary...")
@@ -133,7 +141,9 @@ def main():
     # Update linked issues
     if PR.get("merged"):
         print("PR is merged, labeling fixed issues...")
-        label_fixed_issues(PR["number"])
+        label_fixed_issues(pr_number)
+        print("Removing TODO label from PR...")
+        remove_todos_on_merge(pr_number)
 
 
 if __name__ == "__main__":

{ultralytics_actions-0.0.25 → ultralytics_actions-0.0.26}/ultralytics_actions.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ultralytics-actions
-Version: 0.0.25
+Version: 0.0.26
 Summary: Ultralytics Actions for GitHub automation and PR management.
 Author-email: Glenn Jocher <glenn.jocher@ultralytics.com>
 Maintainer-email: Ultralytics <hello@ultralytics.com>