format

Signed-off-by: harupy <17039389+harupy@users.noreply.github.com>
mlflow · Jul 4, 2024 · 95cfd72 · 95cfd72
1 parent acb31d1
commit 95cfd72
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/tests/store/artifact/test_databricks_artifact_repo.py b/tests/store/artifact/test_databricks_artifact_repo.py
@@ -713,19 +713,19 @@ def test_list_artifacts_handles_pagination(databricks_artifact_repo):
     ) as message_mock, mock.patch(
         f"{DATABRICKS_ARTIFACT_REPOSITORY}._call_endpoint",
         side_effect=list_artifact_paginated_response_protos,
-    ): 
-
+    ):
         artifacts = databricks_artifact_repo.list_artifacts()
         assert {file.path for file in artifacts} == {"a.txt", "b", "c.txt", "d", "e.txt", "f"}
         calls = [
-        mock.call(ListArtifacts(run_id=MOCK_RUN_ID, path="")),
-        mock.call(ListArtifacts(run_id=MOCK_RUN_ID, path="", page_token="2")),
-        mock.call(ListArtifacts(run_id=MOCK_RUN_ID, path="", page_token="4")),
-        mock.call(ListArtifacts(run_id=MOCK_RUN_ID, path="", page_token="6")),
+            mock.call(ListArtifacts(run_id=MOCK_RUN_ID, path="")),
+            mock.call(ListArtifacts(run_id=MOCK_RUN_ID, path="", page_token="2")),
+            mock.call(ListArtifacts(run_id=MOCK_RUN_ID, path="", page_token="4")),
+            mock.call(ListArtifacts(run_id=MOCK_RUN_ID, path="", page_token="6")),
         ]
-        
+
         assert message_mock.mock_calls == calls
 
+
 def test_get_read_credential_infos_handles_pagination(databricks_artifact_repo):
     """
     Verifies that the `get_read_credential_infos` method, which is used to resolve read access