Improved type hints [skip ci]

Author: ankane

pgvector/bit.py

@@ -5,7 +5,7 @@
 
 
 class Bit:
-    def __init__(self, value: bytes | str | list[bool] | np.ndarray) -> None:
+    def __init__(self, value: bytes | str | list[bool] | np.ndarray[tuple[int], np.dtype[np.bool | np.uint8]]) -> None:
         if isinstance(value, bytes):
             self._len = 8 * len(value)
             self._data = value

pgvector/halfvec.py

@@ -4,7 +4,7 @@
 
 
 class HalfVector:
-    def __init__(self, value: list[float] | np.ndarray) -> None:
+    def __init__(self, value: list[float] | np.ndarray[tuple[int], np.dtype[np.floating]]) -> None:
         # asarray still copies if same dtype
         if not isinstance(value, np.ndarray) or value.dtype != '>f2':
             value = np.asarray(value, dtype='>f2')

pgvector/vector.py

@@ -4,7 +4,7 @@
 
 
 class Vector:
-    def __init__(self, value: list[float] | np.ndarray) -> None:
+    def __init__(self, value: list[float] | np.ndarray[tuple[int], np.dtype[np.floating]]) -> None:
         # asarray still copies if same dtype
         if not isinstance(value, np.ndarray) or value.dtype != '>f4':
             value = np.asarray(value, dtype='>f4')