Ensure values are always floats for SparseVector

Author: ankane

pgvector/sparsevec.py

@@ -118,7 +118,7 @@ def _from_sparse(self, arr: sparray | spmatrix, /) -> None:
             self._indices = value.coords[-1].tolist()
         else:
             self._indices = value.col.tolist()
-        self._values = value.data.tolist()
+        self._values = [float(v) for v in value.data]
 
     def _from_dense(self, value: list[float] | ndarray, /) -> None:
         self._dim = len(value)

tests/test_sparse_vector.py

@@ -58,6 +58,7 @@ def test_coo_array(self) -> None:
         vec = SparseVector(arr)
         assert vec.to_list() == [1, 0, 2, 0, 3, 0]
         assert vec.indices() == [0, 2, 4]
+        assert isinstance(vec.values()[0], float)
 
     def test_coo_array_dimensions(self) -> None:
         if np is None or sparse is None: