fix: fix type hints

qdrant · Jan 23, 2025 · 01894a8 · 01894a8
1 parent 7976b88
commit 01894a8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/qdrant_client/embed/embedder.py b/qdrant_client/embed/embedder.py
@@ -28,7 +28,7 @@ class ModelInstance(BaseModel, Generic[T], arbitrary_types_allowed=True):  # typ
 
 
 class Embedder:
-    def __init__(self, threads: Optional[int] = None, **kwargs) -> None:
+    def __init__(self, threads: Optional[int] = None, **kwargs: Any) -> None:
         self.embedding_models: dict[str, list[ModelInstance[TextEmbedding]]] = defaultdict(list)
         self.sparse_embedding_models: dict[str, list[ModelInstance[SparseTextEmbedding]]] = (
             defaultdict(list)

diff --git a/qdrant_client/embed/model_embedder.py b/qdrant_client/embed/model_embedder.py
@@ -39,7 +39,7 @@ def process(self, items: Iterable[tuple[int, Any]]) -> Iterable[tuple[int, Any]]
 class ModelEmbedder:
     MAX_INTERNAL_BATCH_SIZE = 4
 
-    def __init__(self, parser: Optional[ModelSchemaParser] = None, **kwargs):
+    def __init__(self, parser: Optional[ModelSchemaParser] = None, **kwargs: Any):
         self._batch_accumulator: dict[str, list[INFERENCE_OBJECT_TYPES]] = {}
         self._embed_storage: dict[str, list[NumericVector]] = {}
         self._embed_inspector = InspectorEmbed(parser=parser)