Fixed linting issues.

SeldonIO · May 10, 2024 · c2cf03c · c2cf03c
1 parent 66b63c9
commit c2cf03c
Show file tree

Hide file tree

Showing 11 changed files with 1,987 additions and 644 deletions.
diff --git a/...ng/testserver/models/text-model/models.py → ...estserver/models/text-model/text-model.py b/...ng/testserver/models/text-model/models.py → ...estserver/models/text-model/text-model.py
diff --git a/mlserver/batching/hooks.py b/mlserver/batching/hooks.py
@@ -1,5 +1,5 @@
 from functools import wraps
-from typing import Awaitable, Callable, Optional, AsyncIterable
+from typing import Awaitable, Callable, Optional, AsyncIterator
 
 from ..errors import MLServerError
 from ..model import MLModel
@@ -55,7 +55,7 @@ async def _inner(payload: InferenceRequest) -> InferenceResponse:
 
 
 def not_implemented_warning(
-    f: Callable[[AsyncIterable[InferenceResponse]], AsyncIterable[InferenceResponse]],
+    f: Callable[[AsyncIterator[InferenceRequest]], AsyncIterator[InferenceResponse]],
 ):
     """
     Decorator to lets users know that adaptive batching is not required on
@@ -69,8 +69,8 @@ def not_implemented_warning(
 
     @wraps(f)
     async def _inner_stream(
-        payload: InferenceRequest,
-    ) -> AsyncIterable[InferenceResponse]:
+        payload: AsyncIterator[InferenceRequest],
+    ) -> AsyncIterator[InferenceResponse]:
         model = _get_model(f)
         logger.warning(
             warning_template.format(model_name=model.name, f_name=f.__name__)

diff --git a/mlserver/grpc/dataplane_pb2.py b/mlserver/grpc/dataplane_pb2.py