Changes

Author: AryanAhadinia

tests/integration/test_launch_apertus.py

@@ -95,9 +95,7 @@ async def _wait_for_job_running(
         if status == JobStatus.TIMEOUT:
             pytest.fail(f"Job {job_id} timed out before becoming RUNNING.")
         await asyncio.sleep(poll_interval_seconds)
-    pytest.fail(
-        f"Job {job_id} did not reach RUNNING within {timeout_minutes} minutes."
-    )
+    pytest.fail(f"Job {job_id} did not reach RUNNING within {timeout_minutes} minutes.")
 
 
 async def _wait_for_model_healthy(
@@ -134,5 +132,7 @@ async def test_launch_apertus_and_health(launcher, cscs_api_key):
     assert status == JobStatus.RUNNING
 
     # Wait for the model endpoint to become HEALTHY
-    health = await _wait_for_model_healthy(served_model_name, cscs_api_key, health_timeout)
+    health = await _wait_for_model_healthy(
+        served_model_name, cscs_api_key, health_timeout
+    )
     assert health == ModelHealth.HEALTHY