Apply fmt

Author: gkiko10

databricks/sdk/mixins/jobs.py

@@ -1,4 +1,5 @@
 from typing import Optional
+
 from databricks.sdk.service import jobs
 
 

tests/test_jobs_mixin.py

@@ -3,8 +3,12 @@
 
 from databricks.sdk import WorkspaceClient
 
+
 def make_path_pattern(run_id: int, page_token: str) -> str:
-    return re.compile(f'{re.escape("http://localhost/api/")}2.\d{re.escape(f"/jobs/runs/get?page_token={page_token}&run_id={run_id}")}')
+    return re.compile(
+        f'{re.escape("http://localhost/api/")}2.\d{re.escape(f"/jobs/runs/get?page_token={page_token}&run_id={run_id}")}'
+    )
+
 
 def test_get_run_pagination_with_tasks(config, requests_mock):
     run1 = {
@@ -26,12 +30,9 @@ def test_get_run_pagination_with_tasks(config, requests_mock):
         "prev_page_token": "initialToken"
     }
     run3 = {"tasks": [{"run_id": 4}], "next_page_token": None, "prev_page_token": "tokenToSecondPage"}
-    requests_mock.get(make_path_pattern(1337, "initialToken"),
-                      text=json.dumps(run1))
-    requests_mock.get(make_path_pattern(1337, "tokenToSecondPage"),
-                      text=json.dumps(run2))
-    requests_mock.get(make_path_pattern(1337, "tokenToThirdPage"),
-                      text=json.dumps(run3))
+    requests_mock.get(make_path_pattern(1337, "initialToken"), text=json.dumps(run1))
+    requests_mock.get(make_path_pattern(1337, "tokenToSecondPage"), text=json.dumps(run2))
+    requests_mock.get(make_path_pattern(1337, "tokenToThirdPage"), text=json.dumps(run3))
     w = WorkspaceClient(config=config)
 
     run = w.jobs.get_run(1337, page_token="initialToken")
@@ -86,12 +87,9 @@ def test_get_run_pagination_with_iterations(config, requests_mock):
         "next_page_token": None,
         "prev_page_token": "tokenToSecondPage"
     }
-    requests_mock.get(make_path_pattern(1337, "initialToken"),
-                      text=json.dumps(run1))
-    requests_mock.get(make_path_pattern(1337, "tokenToSecondPage"),
-                      text=json.dumps(run2))
-    requests_mock.get(make_path_pattern(1337, "tokenToThirdPage"),
-                      text=json.dumps(run3))
+    requests_mock.get(make_path_pattern(1337, "initialToken"), text=json.dumps(run1))
+    requests_mock.get(make_path_pattern(1337, "tokenToSecondPage"), text=json.dumps(run2))
+    requests_mock.get(make_path_pattern(1337, "tokenToThirdPage"), text=json.dumps(run3))
     w = WorkspaceClient(config=config)
 
     run = w.jobs.get_run(1337, page_token="initialToken")