Improved type hints [skip ci]

Author: ankane

pgvector/bit.py

@@ -31,7 +31,7 @@ def __init__(self, value: Any) -> None:
     def __repr__(self) -> str:
         return f'Bit({self.to_text()})'
 
-    def __eq__(self, other: Any) -> bool:
+    def __eq__(self, other: object) -> bool:
         if isinstance(other, self.__class__):
             return self._len == other._len and self._data == other._data
         return False

pgvector/halfvec.py

@@ -18,7 +18,7 @@ def __init__(self, value: Any) -> None:
     def __repr__(self) -> str:
         return f'HalfVector({self.to_list()})'
 
-    def __eq__(self, other: Any) -> bool:
+    def __eq__(self, other: object) -> bool:
         if isinstance(other, self.__class__):
             return np.array_equal(self.to_numpy(), other.to_numpy())
         return False
@@ -48,7 +48,7 @@ def from_binary(cls, value: bytes) -> HalfVector:
         return cls(np.frombuffer(value, dtype='>f2', count=dim, offset=4))
 
     @classmethod
-    def _to_db(cls, value: Any, dim: int | None = None) -> str | None:
+    def _to_db(cls, value: object, dim: int | None = None) -> str | None:
         if value is None:
             return value
 
@@ -61,7 +61,7 @@ def _to_db(cls, value: Any, dim: int | None = None) -> str | None:
         return value.to_text()
 
     @classmethod
-    def _to_db_binary(cls, value: Any) -> bytes | None:
+    def _to_db_binary(cls, value: object) -> bytes | None:
         if value is None:
             return value
 

pgvector/sparsevec.py

@@ -28,7 +28,7 @@ def __repr__(self) -> str:
         elements = dict(zip(self._indices, self._values))
         return f'SparseVector({elements}, {self._dim})'
 
-    def __eq__(self, other: Any) -> bool:
+    def __eq__(self, other: object) -> bool:
         if isinstance(other, self.__class__):
             return self.dimensions() == other.dimensions() and self.indices() == other.indices() and self.values() == other.values()
         return False
@@ -126,7 +126,7 @@ def _from_parts(cls, dim: int, indices: list[int], values: list[float]) -> Spars
         return vec
 
     @classmethod
-    def _to_db(cls, value: Any, dim: int | None = None) -> str | None:
+    def _to_db(cls, value: object, dim: int | None = None) -> str | None:
         if value is None:
             return value
 
@@ -139,7 +139,7 @@ def _to_db(cls, value: Any, dim: int | None = None) -> str | None:
         return value.to_text()
 
     @classmethod
-    def _to_db_binary(cls, value: Any) -> bytes | None:
+    def _to_db_binary(cls, value: object) -> bytes | None:
         if value is None:
             return value
 

pgvector/vector.py

@@ -18,7 +18,7 @@ def __init__(self, value: Any) -> None:
     def __repr__(self) -> str:
         return f'Vector({self.to_list()})'
 
-    def __eq__(self, other: Any) -> bool:
+    def __eq__(self, other: object) -> bool:
         if isinstance(other, self.__class__):
             return np.array_equal(self.to_numpy(), other.to_numpy())
         return False
@@ -48,7 +48,7 @@ def from_binary(cls, value: bytes) -> Vector:
         return cls(np.frombuffer(value, dtype='>f4', count=dim, offset=4))
 
     @classmethod
-    def _to_db(cls, value: Any, dim: int | None = None) -> str | None:
+    def _to_db(cls, value: object, dim: int | None = None) -> str | None:
         if value is None:
             return value
 
@@ -61,7 +61,7 @@ def _to_db(cls, value: Any, dim: int | None = None) -> str | None:
         return value.to_text()
 
     @classmethod
-    def _to_db_binary(cls, value: Any) -> bytes | None:
+    def _to_db_binary(cls, value: object) -> bytes | None:
         if value is None:
             return value