From ead96a103a45839fe7d3d2d5b41bd586f61c78a4 Mon Sep 17 00:00:00 2001
From: Sergiy Matusevych <sergiym@microsoft.com>
Date: Fri, 12 Jul 2024 15:31:22 -0700
Subject: [PATCH] black formatting updates

---
 .../services/remote/azure/azure_fileshare.py  |  7 ++---
 .../remote/azure/azure_fileshare_test.py      | 27 +++++++++----------
 2 files changed, 16 insertions(+), 18 deletions(-)

diff --git a/mlos_bench/mlos_bench/services/remote/azure/azure_fileshare.py b/mlos_bench/mlos_bench/services/remote/azure/azure_fileshare.py
index 7b2ae0fdd9..4a33bc896a 100644
--- a/mlos_bench/mlos_bench/services/remote/azure/azure_fileshare.py
+++ b/mlos_bench/mlos_bench/services/remote/azure/azure_fileshare.py
@@ -58,7 +58,7 @@ def __init__(
             {
                 "storageAccountName",
                 "storageFileShareName",
-            }
+            },
         )
         self._share_client: Optional[ShareClient] = None
 
@@ -67,8 +67,9 @@ def _get_share_client(self) -> ShareClient:
         Get the Azure file share client object.
         """
         if self._share_client is None:
-            assert self._parent is not None and isinstance(self._parent, SupportsAuth), \
-                "Authorization service not provided. Include service-auth.jsonc?"
+            assert self._parent is not None and isinstance(
+                self._parent, SupportsAuth
+            ), "Authorization service not provided. Include service-auth.jsonc?"
             self._share_client = ShareClient.from_share_url(
                 AzureFileShareService._SHARE_URL.format(
                     account_name=self.config["storageAccountName"],
diff --git a/mlos_bench/mlos_bench/tests/services/remote/azure/azure_fileshare_test.py b/mlos_bench/mlos_bench/tests/services/remote/azure/azure_fileshare_test.py
index 18bfe6425e..f3acb5d77e 100644
--- a/mlos_bench/mlos_bench/tests/services/remote/azure/azure_fileshare_test.py
+++ b/mlos_bench/mlos_bench/tests/services/remote/azure/azure_fileshare_test.py
@@ -28,9 +28,7 @@ def test_download_file(
     local_path = f"{local_folder}/{filename}"
 
     config: dict = {}
-    with patch.object(
-        azure_fileshare, "_share_client"
-    ) as mock_share_client, patch.object(
+    with patch.object(azure_fileshare, "_share_client") as mock_share_client, patch.object(
         mock_share_client, "get_file_client"
     ) as mock_get_file_client, patch.object(
         mock_share_client, "get_directory_client"
@@ -88,9 +86,7 @@ def test_download_folder_non_recursive(
     mock_share_client = azure_fileshare._share_client  # pylint: disable=protected-access
 
     config: dict = {}
-    with patch.object(
-        azure_fileshare, "_share_client"
-    ) as mock_share_client, patch.object(
+    with patch.object(azure_fileshare, "_share_client") as mock_share_client, patch.object(
         mock_share_client, "get_directory_client"
     ) as mock_get_directory_client, patch.object(
         mock_share_client, "get_file_client"
@@ -123,9 +119,7 @@ def test_download_folder_recursive(
     local_folder = "some/local/folder"
     dir_client_returns = make_dir_client_returns(remote_folder)
     config: dict = {}
-    with patch.object(
-        azure_fileshare, "_share_client"
-    ) as mock_share_client, patch.object(
+    with patch.object(azure_fileshare, "_share_client") as mock_share_client, patch.object(
         mock_share_client, "get_directory_client"
     ) as mock_get_directory_client, patch.object(
         mock_share_client, "get_file_client"
@@ -167,8 +161,9 @@ def test_upload_file(
     mock_isdir.return_value = False
 
     config: dict = {}
-    with patch.object(azure_fileshare, "_share_client") as mock_share_client, \
-         patch.object(mock_share_client, "get_file_client") as mock_get_file_client:
+    with patch.object(azure_fileshare, "_share_client") as mock_share_client, patch.object(
+        mock_share_client, "get_file_client"
+    ) as mock_get_file_client:
         azure_fileshare.upload(config, local_path, remote_path)
 
     mock_get_file_client.assert_called_with(remote_path)
@@ -240,8 +235,9 @@ def test_upload_directory_non_recursive(
     config: dict = {}
 
     config: dict = {}
-    with patch.object(azure_fileshare, "_share_client") as mock_share_client, \
-         patch.object(mock_share_client, "get_file_client") as mock_get_file_client:
+    with patch.object(azure_fileshare, "_share_client") as mock_share_client, patch.object(
+        mock_share_client, "get_file_client"
+    ) as mock_get_file_client:
         azure_fileshare.upload(config, local_folder, remote_folder, recursive=False)
 
     mock_get_file_client.assert_called_with(f"{remote_folder}/a_file_1.csv")
@@ -266,8 +262,9 @@ def test_upload_directory_recursive(
     config: dict = {}
 
     config: dict = {}
-    with patch.object(azure_fileshare, "_share_client") as mock_share_client, \
-         patch.object(mock_share_client, "get_file_client") as mock_get_file_client:
+    with patch.object(azure_fileshare, "_share_client") as mock_share_client, patch.object(
+        mock_share_client, "get_file_client"
+    ) as mock_get_file_client:
         azure_fileshare.upload(config, local_folder, remote_folder, recursive=True)
 
     mock_get_file_client.assert_has_calls(