fix: lint

Author: pawelpaszki

tests/e2e/mnist_raycluster_sdk_oauth_test.py

@@ -76,7 +76,7 @@ def run_mnist_raycluster_sdk_oauth(self):
 
     def assert_jobsubmit_withoutLogin(self, cluster):
         dashboard_url = cluster.cluster_dashboard_uri()
-        
+
         # Verify that job submission is actually blocked by attempting to submit without auth
         # The endpoint path depends on whether we're using HTTPRoute (with path prefix) or not
         if "/ray/" in dashboard_url:
@@ -87,7 +87,7 @@ def assert_jobsubmit_withoutLogin(self, cluster):
             # OpenShift Route format: https://hostname
             # API endpoint is directly under the hostname
             api_url = dashboard_url + "/api/jobs/"
-        
+
         jobdata = {
             "entrypoint": "python mnist.py",
             "runtime_env": {
@@ -96,20 +96,22 @@ def assert_jobsubmit_withoutLogin(self, cluster):
                 "env_vars": get_setup_env_variables(),
             },
         }
-        
+
         # Try to submit a job without authentication
         # Follow redirects to see the final response - if it redirects to login, that's still a failure
-        response = requests.post(api_url, verify=False, json=jobdata, allow_redirects=True)
-        
+        response = requests.post(
+            api_url, verify=False, json=jobdata, allow_redirects=True
+        )
+
         # Check if the submission was actually blocked
         # Success indicators that submission was blocked:
         # 1. Status code 403 (Forbidden)
         # 2. Status code 302 (Redirect to login) - but we need to verify the final response after redirect
         # 3. Status code 200 but with HTML content (login page) instead of JSON (job submission response)
         # 4. Status code 401 (Unauthorized)
-        
+
         submission_blocked = False
-        
+
         if response.status_code == 403:
             submission_blocked = True
         elif response.status_code == 401:
@@ -120,8 +122,8 @@ def assert_jobsubmit_withoutLogin(self, cluster):
             submission_blocked = True  # Redirect to login means submission failed
         elif response.status_code == 200:
             # Check if response is HTML (login page) instead of JSON (job submission response)
-            content_type = response.headers.get('Content-Type', '')
-            if 'text/html' in content_type or 'application/json' not in content_type:
+            content_type = response.headers.get("Content-Type", "")
+            if "text/html" in content_type or "application/json" not in content_type:
                 # Got HTML (likely login page) instead of JSON - submission was blocked
                 submission_blocked = True
             else:
@@ -130,7 +132,7 @@ def assert_jobsubmit_withoutLogin(self, cluster):
                     json_response = response.json()
                     # If it's a successful job submission, it should have a 'job_id' or 'submission_id'
                     # If it's an error, it might have 'error' or 'message'
-                    if 'job_id' in json_response or 'submission_id' in json_response:
+                    if "job_id" in json_response or "submission_id" in json_response:
                         # Job was actually submitted - this is a failure!
                         submission_blocked = False
                     else:
@@ -139,20 +141,28 @@ def assert_jobsubmit_withoutLogin(self, cluster):
                 except ValueError:
                     # Not JSON - likely HTML login page
                     submission_blocked = True
-        
+
         if not submission_blocked:
-            assert False, f"Job submission succeeded without authentication! Status: {response.status_code}, Response: {response.text[:200]}"
-        
+            assert (
+                False
+            ), f"Job submission succeeded without authentication! Status: {response.status_code}, Response: {response.text[:200]}"
+
         # Also verify that RayJobClient cannot be used without authentication
         try:
             client = RayJobClient(address=dashboard_url, verify=False)
             # Try to call a method to trigger the connection and authentication check
             client.list_jobs()
-            assert False, "RayJobClient succeeded without authentication - this should not be possible"
-        except (requests.exceptions.JSONDecodeError, requests.exceptions.HTTPError, Exception):
+            assert (
+                False
+            ), "RayJobClient succeeded without authentication - this should not be possible"
+        except (
+            requests.exceptions.JSONDecodeError,
+            requests.exceptions.HTTPError,
+            Exception,
+        ):
             # Any exception is expected when trying to use the client without auth
             pass
-        
+
         assert True, "Job submission without authentication was correctly blocked"
 
     def assert_jobsubmit_withlogin(self, cluster):
@@ -165,13 +175,18 @@ def assert_jobsubmit_withlogin(self, cluster):
         # This ensures that the authentication check in assert_jobsubmit_withoutLogin actually worked
         existing_jobs = client.list_jobs()
         if existing_jobs:
-            job_ids = [job.job_id if hasattr(job, 'job_id') else str(job) for job in existing_jobs]
+            job_ids = [
+                job.job_id if hasattr(job, "job_id") else str(job)
+                for job in existing_jobs
+            ]
             assert False, (
                 f"Found {len(existing_jobs)} existing job(s) before authenticated submission: {job_ids}. "
                 "This indicates that the unauthenticated job submission test failed to properly block submission."
             )
         else:
-            print("Verified: No jobs exist from the previous unauthenticated submission attempt.")
+            print(
+                "Verified: No jobs exist from the previous unauthenticated submission attempt."
+            )
 
         submission_id = client.submit_job(
             entrypoint="python mnist.py",