Improved code [skip ci]

ankane · ankane · commit 1aec1c0e17a0 · 2024-05-23T15:07:31.000-04:00
diff --git a/pgvector/utils/bit.py b/pgvector/utils/bit.py
@@ -13,6 +13,12 @@ def __init__(self, value):
         else:
             self.value = np.array(value, dtype=bool)
 
+    def __str__(self):
+        return self.__class__.to_db(self)
+
+    def __repr__(self):
+        return f'Bit({self})'
+
     def to_db(value):
         if not isinstance(value, Bit):
             raise ValueError('expected bit')
@@ -26,9 +32,3 @@ def to_db_binary(value):
 
         value = value.value
         return pack('>i', len(value)) + np.packbits(value).tobytes()
-
-    def __str__(self):
-        return self.__class__.to_db(self)
-
-    def __repr__(self):
-        return f'Bit({self})'
diff --git a/pgvector/utils/halfvec.py b/pgvector/utils/halfvec.py
@@ -15,6 +15,9 @@ def __init__(self, value):
     def to_list(self):
         return list(self.value)
 
+    def __repr__(self):
+        return f'HalfVector({self.value})'
+
     def to_db(value, dim=None):
         if value is None:
             return value
@@ -43,6 +46,3 @@ def from_db_binary(value):
             return value
         dim, unused = unpack_from('>HH', value)
         return HalfVector(unpack_from(f'>{dim}e', value, 4))
-
-    def __repr__(self):
-        return f'HalfVector({self.value})'
diff --git a/pgvector/utils/sparsevec.py b/pgvector/utils/sparsevec.py
@@ -22,6 +22,9 @@ def to_dense(self):
             vec[i] = v
         return vec
 
+    def __repr__(self):
+        return f'SparseVector({self.dim}, {self.indices}, {self.values})'
+
     def to_db(value, dim=None):
         if value is None:
             return value
@@ -68,6 +71,3 @@ def from_db_binary(value):
         indices = unpack_from(f'>{nnz}i', value, 12)
         values = unpack_from(f'>{nnz}f', value, 12 + nnz * 4)
         return SparseVector(int(dim), indices, values)
-
-    def __repr__(self):
-        return f'SparseVector({self.dim}, {self.indices}, {self.values})'
