Skip to content

Commit b5d9b70

Browse files
committed
Improved _from_dense method
1 parent 063995b commit b5d9b70

1 file changed

Lines changed: 7 additions & 2 deletions

File tree

‎pgvector/sparsevec.py‎

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -137,8 +137,13 @@ def _from_sparse(self, arr: sparray | spmatrix, /) -> None:
137137

138138
def _from_dense(self, value: list[float] | ndarray, /) -> None:
139139
self._dim = len(value)
140-
self._indices = [i for i, v in enumerate(value) if v != 0]
141-
self._values = [float(value[i]) for i in self._indices]
140+
self._indices = []
141+
self._values = []
142+
for i, v in enumerate(value):
143+
v = float(v)
144+
if v != 0:
145+
self._indices.append(i)
146+
self._values.append(v)
142147

143148
def _validate(self) -> None:
144149
if self._dim < 0:

0 commit comments

Comments
 (0)