Removed warning for result of np.unpackbits

Author: ankane

pgvector/bit.py

@@ -15,7 +15,9 @@ def __init__(self, value):
                 value = np.asarray(value)
 
                 if value.dtype != np.bool:
-                    warn('expected elements to be boolean', stacklevel=2)
+                    # allow result of np.unpackbits
+                    if value.dtype != np.uint8 or np.any(value > 1):
+                        warn('expected elements to be boolean', stacklevel=2)
                     value = value.astype(bool)
 
                 if value.ndim != 1:

tests/test_bit.py

@@ -30,6 +30,10 @@ def test_ndarray(self):
         assert Bit(arr).to_list() == [True, False, True]
         assert np.array_equal(Bit(arr).to_numpy(), arr)
 
+    def test_ndarray_unpackbits(self):
+        arr = np.unpackbits(np.array([254, 7, 0], dtype=np.uint8))
+        assert Bit(arr).to_text() == '111111100000011100000000'
+
     def test_ndarray_uint8(self):
         arr = np.array([254, 7, 0], dtype=np.uint8)
         with pytest.warns(UserWarning, match='expected elements to be boolean'):