Skip to content

Commit eeaa6cb

Browse files
authored
[hnswlib] Fix Index.get_items return types (#16578)
1 parent 4c6a006 commit eeaa6cb

1 file changed

Lines changed: 5 additions & 3 deletions

File tree

‎stubs/hnswlib/hnswlib.pyi‎

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -25,13 +25,15 @@ class Index:
2525
def get_ids_list(self) -> list[int]: ...
2626

2727
@overload
28-
def get_items(self, ids: ArrayLike | None = ..., return_type: Literal["list"] = ...) -> list[float]: ...
28+
def get_items(self, ids: ArrayLike | None = None, return_type: Literal["numpy"] = "numpy") -> NDArray[np.float32]: ...
2929
@overload
30-
def get_items(self, ids: ArrayLike | None = ..., return_type: Literal["numpy"] = ...) -> NDArray[np.float32]: ...
30+
def get_items(self, ids: ArrayLike | None, return_type: Literal["list"]) -> list[list[float]]: ...
31+
@overload
32+
def get_items(self, ids: ArrayLike | None = None, *, return_type: Literal["list"]) -> list[list[float]]: ...
3133
@overload
3234
def get_items(
3335
self, ids: ArrayLike | None = None, return_type: Literal["numpy", "list"] = "numpy"
34-
) -> NDArray[np.float32] | list[float]: ...
36+
) -> NDArray[np.float32] | list[list[float]]: ...
3537

3638
def get_max_elements(self) -> int: ...
3739
def index_file_size(self) -> int: ...

0 commit comments

Comments
 (0)