style: format

Author: racinmat

tests/unit/aiplatform/test_pipeline_job_schedules.py

@@ -438,7 +438,7 @@ def teardown_method(self):
     def test_block_until_complete_logs_symbolic_state_name(self):
         """State log must use symbolic enum name, not a bare integer (regression for Python 3.11+)."""
         state_sequence = [
-            gca_schedule.Schedule.State.ACTIVE,     # first loop check
+            gca_schedule.Schedule.State.ACTIVE,  # first loop check
             gca_schedule.Schedule.State.COMPLETED,  # second check exits loop
         ]
         state_index = [0]
@@ -461,17 +461,18 @@ def get_state():
 
         # time.time: first call sets previous_time=0; second gives 10 → triggers log (10 >= 5)
         time_vals = iter([0.0, 10.0, 20.0])
-        with mock.patch("google.cloud.aiplatform.schedules.time.time", side_effect=time_vals), \
-             mock.patch("google.cloud.aiplatform.schedules.time.sleep"), \
-             mock.patch.object(
-                 aiplatform_schedules._LOGGER, "info",
-                 side_effect=lambda msg, *a, **kw: logged_messages.append(msg)
-             ):
+        with mock.patch(
+            "google.cloud.aiplatform.schedules.time.time", side_effect=time_vals
+        ), mock.patch(
+            "google.cloud.aiplatform.schedules.time.sleep"
+        ), mock.patch.object(
+            aiplatform_schedules._LOGGER,
+            "info",
+            side_effect=lambda msg, *a, **kw: logged_messages.append(msg),
+        ):
             aiplatform_schedules._Schedule._block_until_complete(mock_schedule)
 
-        state_log = next(
-            (m for m in logged_messages if "current state" in m), None
-        )
+        state_log = next((m for m in logged_messages if "current state" in m), None)
         assert state_log is not None, "No 'current state' log message found"
         assert "ACTIVE" in state_log
         assert "current state:\n1" not in state_log

tests/unit/aiplatform/test_pipeline_jobs.py

@@ -726,11 +726,13 @@ def test_block_until_complete_logs_symbolic_state_name(
 
         logged_messages = []
 
-        with patch.object(storage.Blob, "download_as_bytes") as mock_load, \
-             mock.patch.object(
-                 pipeline_jobs._LOGGER, "info",
-                 side_effect=lambda msg, *a, **kw: logged_messages.append(msg)
-             ):
+        with patch.object(
+            storage.Blob, "download_as_bytes"
+        ) as mock_load, mock.patch.object(
+            pipeline_jobs._LOGGER,
+            "info",
+            side_effect=lambda msg, *a, **kw: logged_messages.append(msg),
+        ):
             mock_load.return_value = _TEST_PIPELINE_SPEC_JSON.encode()
 
             job = pipeline_jobs.PipelineJob(
@@ -740,9 +742,7 @@ def test_block_until_complete_logs_symbolic_state_name(
             )
             job.run(sync=True, create_request_timeout=None)
 
-        state_log = next(
-            (m for m in logged_messages if "current state" in m), None
-        )
+        state_log = next((m for m in logged_messages if "current state" in m), None)
         assert state_log is not None, "No 'current state' log message found"
         assert "PIPELINE_STATE_RUNNING" in state_log
         assert "current state:\n3" not in state_log

tests/unit/aiplatform/test_training_jobs.py

@@ -1307,7 +1307,9 @@ def test_block_until_complete_logs_symbolic_state_name(
         ), mock.patch.object(
             training_jobs, "_JOB_WAIT_TIME", 0
         ), mock.patch.object(
-            training_jobs._LOGGER, "info", side_effect=lambda msg, *a, **kw: logged_messages.append(msg)
+            training_jobs._LOGGER,
+            "info",
+            side_effect=lambda msg, *a, **kw: logged_messages.append(msg),
         ):
             mock_pkg.return_value = _TEST_OUTPUT_PYTHON_PACKAGE_PATH
             mock_create.return_value = gca_training_pipeline.TrainingPipeline(
@@ -1334,9 +1336,7 @@ def test_block_until_complete_logs_symbolic_state_name(
             )
             job.run(base_output_dir=_TEST_BASE_OUTPUT_DIR, sync=True)
 
-        state_log = next(
-            (m for m in logged_messages if "current state" in m), None
-        )
+        state_log = next((m for m in logged_messages if "current state" in m), None)
         assert state_log is not None, "No 'current state' log message found"
         assert "PIPELINE_STATE_RUNNING" in state_log
         assert "current state:\n3" not in state_log