ruff

Author: bpblanken

v03_pipeline/lib/tasks/dataproc/create_dataproc_cluster_test.py

@@ -50,7 +50,7 @@ def test_spinup_cluster_already_exists_failed(
         mock_client = mock_cluster_controller.return_value
         mock_client.get_cluster.return_value = SimpleNamespace(
             status=SimpleNamespace(
-                state=google.cloud.dataproc_v1.types.clusters.ClusterStatus.State.ERROR
+                state=google.cloud.dataproc_v1.types.clusters.ClusterStatus.State.ERROR,
             ),
             cluster_name='abc',
         )
@@ -75,7 +75,7 @@ def test_spinup_cluster_already_exists_success(
         mock_client = mock_cluster_controller.return_value
         mock_client.get_cluster.return_value = SimpleNamespace(
             status=SimpleNamespace(
-                state=google.cloud.dataproc_v1.types.clusters.ClusterStatus.State.RUNNING
+                state=google.cloud.dataproc_v1.types.clusters.ClusterStatus.State.RUNNING,
             ),
         )
         mock_client.create_cluster.side_effect = (

v03_pipeline/lib/tasks/dataproc/run_pipeline_on_dataproc_test.py

@@ -68,7 +68,9 @@ def test_job_already_exists_success(
         mock_create_dataproc_cluster.return_value = MockCompleteTask()
         mock_client = mock_job_controller_client.return_value
         mock_client.get_job.return_value = SimpleNamespace(
-            status=SimpleNamespace(state=google.cloud.dataproc_v1.types.jobs.JobStatus.State.DONE),
+            status=SimpleNamespace(
+                state=google.cloud.dataproc_v1.types.jobs.JobStatus.State.DONE
+            ),
         )
         worker = luigi.worker.Worker()
         task = RunPipelineOnDataprocTask(
@@ -136,7 +138,9 @@ def test_job_success(
                 'job not found',
             ),
             SimpleNamespace(
-                status=SimpleNamespace(state=google.cloud.dataproc_v1.types.jobs.JobStatus.State.DONE),
+                status=SimpleNamespace(
+                    state=google.cloud.dataproc_v1.types.jobs.JobStatus.State.DONE
+                ),
             ),
         ]
         operation = mock_client.submit_job_as_operation.return_value