Git rebase on torch main

Author: jackzhxng

.github/scripts/check_labels.py

@@ -51,18 +51,18 @@ def main() -> None:
 
     try:
         if not has_required_labels(pr):
-            print(LABEL_ERR_MSG)
+            print(LABEL_ERR_MSG, flush=True)
             add_label_err_comment(pr)
             if args.exit_non_zero:
-                sys.exit(1)
+                raise RuntimeError("PR does not have required labels")
         else:
             print("pass")
             delete_all_label_err_comments(pr)
     except Exception as e:
         print("general exception")
         print(e)
         if args.exit_non_zero:
-            sys.exit(1)
+            raise RuntimeError(f"Error checking labels: {e}") from e
 
     print("success")
     sys.exit(0)

.github/scripts/github_utils.py

@@ -66,17 +66,29 @@ def gh_fetch_url_and_headers(
         raise
 
 
+# def gh_fetch_url(
+#     url: str,
+#     *,
+#     headers: Optional[Dict[str, str]] = None,
+#     data: Union[Optional[Dict[str, Any]], str] = None,
+#     method: Optional[str] = None,
+#     reader: Callable[[Any], Any] = lambda x: x.read(),
+# ) -> Any:
+#     print(f"api return conent: {gh_fetch_url_and_headers(url, headers=headers, data=data, reader=json.load, method=method)}")
+#     return gh_fetch_url_and_headers(
+#         url, headers=headers, data=data, reader=json.load, method=method
+#     )[1]
+
 def gh_fetch_url(
     url: str,
     *,
     headers: Optional[Dict[str, str]] = None,
     data: Union[Optional[Dict[str, Any]], str] = None,
     method: Optional[str] = None,
-    reader: Callable[[Any], Any] = lambda x: x.read(),
+    reader: Callable[[Any], Any] = json.load,
 ) -> Any:
-    print(f"api return conent: {gh_fetch_url_and_headers(url, headers=headers, data=data, reader=json.load, method=method)}")
     return gh_fetch_url_and_headers(
-        url, headers=headers, data=data, reader=json.load, method=method
+        url, headers=headers, data=data, reader=reader, method=method
     )[1]
 
 
@@ -168,11 +180,15 @@ def gh_post_commit_comment(
     )
 
 
+# def gh_delete_comment(org: str, repo: str, comment_id: int) -> None:
+#     print("deleting comment")
+#     url = f"{GITHUB_API_URL}/repos/{org}/{repo}/issues/comments/{comment_id}"
+#     print(f"url: {url}")
+#     gh_fetch_url(url, method="DELETE")
+
 def gh_delete_comment(org: str, repo: str, comment_id: int) -> None:
-    print("deleting comment")
     url = f"{GITHUB_API_URL}/repos/{org}/{repo}/issues/comments/{comment_id}"
-    print(f"url: {url}")
-    gh_fetch_url(url, method="DELETE")
+    gh_fetch_url(url, method="DELETE", reader=lambda x: x.read())
 
 
 def gh_fetch_merge_base(org: str, repo: str, base: str, head: str) -> str: