Improved type hints [skip ci]

ankane · ankane · commit 66c4515858ae · 2026-06-02T10:19:23.000-07:00
diff --git a/pgvector/bit.py b/pgvector/bit.py
@@ -1,12 +1,11 @@
 from __future__ import annotations
 import numpy as np
 from struct import pack, unpack_from
-from typing import Any
 from warnings import warn
 
 
 class Bit:
-    def __init__(self, value: Any) -> None:
+    def __init__(self, value: object) -> None:
         if isinstance(value, bytes):
             self._len = 8 * len(value)
             self._data = value
@@ -16,12 +15,18 @@ def __init__(self, value: Any) -> None:
             else:
                 value = np.asarray(value)
 
+                # for mypy
+                assert isinstance(value, np.ndarray)
+
                 if value.dtype != np.bool:
                     # skip warning for 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)
 
+                    # for mypy
+                    assert isinstance(value, np.ndarray)
+
                 if value.ndim != 1:
                     raise ValueError('expected ndim to be 1')
 
