Fix generic type in pynndescent

MinishLab · Nov 13, 2024 · 21912f2 · 21912f2
1 parent c4061b0
commit 21912f2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/vicinity/backends/base.py b/vicinity/backends/base.py
@@ -33,7 +33,7 @@ def dict(self) -> dict[str, Any]:
 
 
 class AbstractBackend(ABC, Generic[ArgType]):
-    argument_class: type[BaseArgs]
+    argument_class: type[ArgType]
 
     def __init__(self, arguments: ArgType, *args: Any, **kwargs: Any) -> None:
         """Initialize the backend with vectors."""

diff --git a/vicinity/backends/pynndescent.py b/vicinity/backends/pynndescent.py
@@ -23,7 +23,7 @@ class PyNNDescentArgs(BaseArgs):
     ] = "cosine"
 
 
-class PyNNDescentBackend(AbstractBackend):
+class PyNNDescentBackend(AbstractBackend[PyNNDescentArgs]):
     argument_class = PyNNDescentArgs
 
     def __init__(