Added casting [skip ci]

ankane · ankane · commit 6c98554e1306 · 2024-06-01T20:13:31.000-07:00
diff --git a/pgvector/utils/halfvec.py b/pgvector/utils/halfvec.py
@@ -26,7 +26,7 @@ def to_numpy(self):
         return self._value
 
     def to_text(self):
-        return '[' + ','.join([str(v) for v in self._value]) + ']'
+        return '[' + ','.join([str(float(v)) for v in self._value]) + ']'
 
     def to_binary(self):
         return pack('>HH', self.dim(), 0) + self._value.tobytes()
diff --git a/pgvector/utils/sparsevec.py b/pgvector/utils/sparsevec.py
@@ -45,7 +45,7 @@ def to_numpy(self):
         return vec
 
     def to_text(self):
-        return '{' + ','.join([f'{i + 1}:{v}' for i, v in zip(self._indices, self._values)]) + '}/' + str(self._dim)
+        return '{' + ','.join([f'{int(i) + 1}:{float(v)}' for i, v in zip(self._indices, self._values)]) + '}/' + str(int(self._dim))
 
     def to_binary(self):
         nnz = len(self._indices)
diff --git a/pgvector/utils/vector.py b/pgvector/utils/vector.py
@@ -26,7 +26,7 @@ def to_numpy(self):
         return self._value
 
     def to_text(self):
-        return '[' + ','.join([str(v) for v in self._value]) + ']'
+        return '[' + ','.join([str(float(v)) for v in self._value]) + ']'
 
     def to_binary(self):
         return pack('>HH', self.dim(), 0) + self._value.tobytes()
