Skip to content

Commit 1913c84

Browse files
committed
Improved type hints [skip ci]
1 parent e86d7c4 commit 1913c84

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

pgvector/sparsevec.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -66,7 +66,7 @@ def to_list(self) -> list[float]:
6666
vec[i] = v
6767
return vec
6868

69-
def to_numpy(self) -> np.ndarray:
69+
def to_numpy(self) -> np.ndarray[tuple[int], np.dtype[np.float32]]:
7070
vec = np.repeat(0.0, self._dim).astype(np.float32)
7171
for i, v in zip(self._indices, self._values):
7272
vec[i] = v

0 commit comments

Comments
 (0)