style: lint fix

Author: snus-kin

.vscode/settings.json

@@ -4,4 +4,4 @@
     ],
     "python.testing.unittestEnabled": false,
     "python.testing.pytestEnabled": true
-}
+}

src/resolver_athena_client/client/athena_client.py

@@ -441,7 +441,7 @@ async def shutdown_worker(
                     for result in results
                     if isinstance(
                         result,
-                        (asyncio.CancelledError, ConnectionError, OSError),
+                        asyncio.CancelledError | ConnectionError | OSError,
                     )
                 ]
 

tests/functional/conftest.py

@@ -12,9 +12,6 @@
 import pytest_asyncio
 from dotenv import load_dotenv
 from grpc.aio import Channel
-from resolver_athena_client.generated.athena.models_pb2 import (
-    ClassificationOutput,
-)
 
 from resolver_athena_client.client.athena_client import AthenaClient
 from resolver_athena_client.client.athena_options import AthenaOptions
@@ -28,6 +25,9 @@
     MAX_DEPLOYMENT_ID_LENGTH,
 )
 from resolver_athena_client.client.models.input_model import ImageData
+from resolver_athena_client.generated.athena.models_pb2 import (
+    ClassificationOutput,
+)
 
 
 def _create_base_test_image_opencv(width: int, height: int) -> np.ndarray:
@@ -216,7 +216,9 @@ async def send(self, image_data: ImageData) -> ClassificationOutput:
 
         if image_data.correlation_id is None:
             image_data.correlation_id = str(uuid.uuid4())
-        future = asyncio.get_event_loop().create_future()
+        future: Future[ClassificationOutput] = (
+            asyncio.get_event_loop().create_future()
+        )
         self._pending_results[image_data.correlation_id] = future
 
         await self._request_queue.put(image_data)

tests/functional/e2e/test_classify_single.py

@@ -14,10 +14,14 @@
 FP_ERROR_TOLERANCE = 1e-4
 
 
+def _test_id(tc: AthenaTestCase) -> str:
+    return str(tc.id)
+
+
 @pytest.mark.asyncio(loop_scope="session")
 @pytest.mark.functional
 @pytest.mark.e2e
-@pytest.mark.parametrize("test_case", TEST_CASES, ids=lambda tc: tc.id)
+@pytest.mark.parametrize("test_case", TEST_CASES, ids=_test_id)
 async def test_e2e_case(
     streaming_sender: StreamingSender,
     test_case: AthenaTestCase,