format

apache · Jan 23, 2025 · 1fe14b5 · 1fe14b5
1 parent 6b989c1
commit 1fe14b5
Showing 1 changed file with 4 additions and 8 deletions.
diff --git a/providers/tests/cncf/kubernetes/operators/test_spark_kubernetes.py b/providers/tests/cncf/kubernetes/operators/test_spark_kubernetes.py
@@ -711,7 +711,8 @@ def test_find_custom_pod_labels(
         mock_await_pod_start,
         mock_await_pod_completion,
         mock_fetch_requested_container_logs,
-        data_file,):
+        data_file,
+    ):
         task_name = "test_find_custom_pod_labels"
         job_spec = yaml.safe_load(data_file("spark/application_template.yaml").read_text())
 
@@ -724,14 +725,9 @@ def test_find_custom_pod_labels(
         )
         context = create_context(op)
         op.execute(context)
-        label_selector = (
-            op._build_find_pod_label_selector(context)
-            + ",spark-role=driver"
-        )
+        label_selector = op._build_find_pod_label_selector(context) + ",spark-role=driver"
         op.find_spark_job(context)
-        mock_get_kube_client.list_namespaced_pod.assert_called_with(
-            'default', label_selector=label_selector
-        )
+        mock_get_kube_client.list_namespaced_pod.assert_called_with("default", label_selector=label_selector)
 
 
 @pytest.mark.db_test