Removed from_binary path from Bit constructor [skip ci]

ankane · ankane · commit 33ec0f2ff007 · 2024-06-01T19:39:25.000-07:00
diff --git a/pgvector/utils/bit.py b/pgvector/utils/bit.py
@@ -4,9 +4,7 @@
 
 class Bit:
     def __init__(self, value):
-        if isinstance(value, bytes):
-            self._value = __class__.from_binary(value)._value
-        elif isinstance(value, str):
+        if isinstance(value, str):
             self._value = __class__.from_text(value)._value
         else:
             value = np.asarray(value, dtype=bool)
diff --git a/tests/test_psycopg.py b/tests/test_psycopg.py
@@ -91,8 +91,8 @@ def test_bit(self):
     def test_bit_binary_format(self):
         embedding = Bit([False, True, False, True, False, False, False, False, True])
         res = conn.execute('SELECT %b::bit(9)', (embedding,), binary=True).fetchone()[0]
-        assert str(Bit(res)) == '010100001'
-        assert repr(Bit(res)) == 'Bit(010100001)'
+        assert str(Bit.from_binary(res)) == '010100001'
+        assert repr(Bit.from_binary(res)) == 'Bit(010100001)'
 
     def test_bit_text_format(self):
         embedding = Bit([False, True, False, True, False, False, False, False, True])
