Update

[ghstack-poisoned]

Author: huydhn

aws/lambda/gha-log-uploader/lambda_function.py

@@ -34,7 +34,9 @@
 # Base64-encoded PEM, the same encoding torchci uses for its app key
 GITHUB_APP_PRIVATE_KEY = os.environ.get("GITHUB_APP_PRIVATE_KEY")
 BUCKET_NAME = "ossci-raw-job-status"
-LOG_CLASSIFIER_URL = "https://vwg52br27lx5oymv4ouejwf4re0akoeg.lambda-url.us-east-1.on.aws"
+LOG_CLASSIFIER_URL = (
+    "https://vwg52br27lx5oymv4ouejwf4re0akoeg.lambda-url.us-east-1.on.aws"
+)
 
 GITHUB_API_URL = "https://api.github.com"
 # Installation tokens last an hour. Refresh early so a warm invocation never

aws/lambda/gha-log-uploader/test_lambda_function.py

@@ -2,12 +2,7 @@
 from unittest.mock import MagicMock, patch
 
 import lambda_function
-from lambda_function import (
-    classify_log,
-    download_log,
-    installation_token,
-    parse_event,
-)
+from lambda_function import classify_log, download_log, installation_token, parse_event
 
 
 def make_response(status_code, content=b"log data", headers=None):
@@ -296,11 +291,10 @@ def test_a_github_blip_propagates_so_lambda_retries(self, s3):
             lambda_function,
             "download_log",
             side_effect=lambda_function.requests.RequestException("boom"),
-        ):
-            with self.assertRaises(lambda_function.requests.RequestException):
-                lambda_function.lambda_handler(
-                    {"repo": "pytorch/pytorch", "job_id": 5}, None
-                )
+        ), self.assertRaises(lambda_function.requests.RequestException):
+            lambda_function.lambda_handler(
+                {"repo": "pytorch/pytorch", "job_id": 5}, None
+            )
 
 
 if __name__ == "__main__":