blacken files

Author: Coniferish

_test_unstructured_client/test__decorators.py

@@ -11,29 +11,32 @@
 @pytest.mark.parametrize(
     ("server_url"),
     [
-        ("https://unstructured-000mock.api.unstructuredapp.io"), # correct url
+        ("https://unstructured-000mock.api.unstructuredapp.io"),  # correct url
         ("unstructured-000mock.api.unstructuredapp.io"),
         ("http://unstructured-000mock.api.unstructuredapp.io/general/v0/general"),
         ("https://unstructured-000mock.api.unstructuredapp.io/general/v0/general"),
         ("unstructured-000mock.api.unstructuredapp.io/general/v0/general"),
-    ]
+    ],
 )
 def test_clean_server_url_on_paid_api_url(server_url: str):
     client = UnstructuredClient(
         server_url=server_url,
         api_key_auth=FAKE_KEY,
     )
-    assert client.general.sdk_configuration.server_url == "https://unstructured-000mock.api.unstructuredapp.io"
+    assert (
+        client.general.sdk_configuration.server_url
+        == "https://unstructured-000mock.api.unstructuredapp.io"
+    )
 
 
 @pytest.mark.parametrize(
     ("server_url"),
     [
-        ("http://localhost:8000"), # correct url
+        ("http://localhost:8000"),  # correct url
         ("localhost:8000"),
         ("localhost:8000/general/v0/general"),
         ("http://localhost:8000/general/v0/general"),
-    ]
+    ],
 )
 def test_clean_server_url_on_localhost(server_url: str):
     client = UnstructuredClient(
@@ -56,15 +59,18 @@ def test_clean_server_url_on_empty_string():
     [
         ("https://unstructured-000mock.api.unstructuredapp.io"),
         ("unstructured-000mock.api.unstructuredapp.io/general/v0/general"),
-    ]
+    ],
 )
 def test_clean_server_url_with_positional_arguments(server_url: str):
     client = UnstructuredClient(
         FAKE_KEY,
         "",
         server_url,
     )
-    assert client.general.sdk_configuration.server_url == "https://unstructured-000mock.api.unstructuredapp.io"
+    assert (
+        client.general.sdk_configuration.server_url
+        == "https://unstructured-000mock.api.unstructuredapp.io"
+    )
 
 
 def test_suggest_defining_url_if_401():
@@ -73,11 +79,11 @@ def test_suggest_defining_url_if_401():
         client = UnstructuredClient(
             api_key_auth=FAKE_KEY,
         )
-        
+
         filename = "_sample_docs/layout-parser-paper-fast.pdf"
-        
+
         with open(filename, "rb") as f:
-            files=shared.Files(
+            files = shared.Files(
                 content=f.read(),
                 file_name=filename,
             )

src/unstructured_client/utils/_decorators.py

@@ -45,25 +45,29 @@ def wrapper(*args: _P.args, **kwargs: _P.kwargs) -> None:
             if url_is_in_kwargs:
                 kwargs["server_url"] = urlunparse(cleaned_url)
             else:
-                args = args[:SERVER_URL_ARG_IDX] + (urlunparse(cleaned_url),) + args[SERVER_URL_ARG_IDX+1:] # type: ignore
-        
+                args = args[:SERVER_URL_ARG_IDX] + (urlunparse(cleaned_url),) + args[SERVER_URL_ARG_IDX + 1 :]  # type: ignore
+
         return func(*args, **kwargs)
 
     return wrapper
 
 
-def suggest_defining_url_if_401(func: Callable[_P, operations.PartitionResponse]) -> Callable[_P, operations.PartitionResponse]:
+def suggest_defining_url_if_401(
+    func: Callable[_P, operations.PartitionResponse]
+) -> Callable[_P, operations.PartitionResponse]:
 
     @functools.wraps(func)
     def wrapper(*args: _P.args, **kwargs: _P.kwargs) -> operations.PartitionResponse:
         try:
             return func(*args, **kwargs)
         except errors.SDKError as e:
             if e.status_code == 401:
-                general_obj: General = args[0] # type: ignore
+                general_obj: General = args[0]  # type: ignore
                 if not general_obj.sdk_configuration.server_url:
-                    warnings.warn("If intending to use the paid API, please define `server_url` in your request.")
-            
+                    warnings.warn(
+                        "If intending to use the paid API, please define `server_url` in your request."
+                    )
+
             return func(*args, **kwargs)
 
-    return wrapper
+    return wrapper