Improved type hints [skip ci]

Author: ankane

pgvector/sqlalchemy/bit.py

@@ -1,6 +1,6 @@
 from sqlalchemy.dialects.postgresql.base import ischema_names
 from sqlalchemy.types import UserDefinedType, Float
-from sqlalchemy import Dialect
+from sqlalchemy import Dialect, Operators
 from typing import Any
 
 
@@ -29,10 +29,10 @@ def process(value):
             return super().bind_processor(dialect)
 
     class comparator_factory(UserDefinedType.Comparator):
-        def hamming_distance(self, other: Any) -> Any:
+        def hamming_distance(self, other: Any) -> Operators:
             return self.op('<~>', return_type=Float)(other)
 
-        def jaccard_distance(self, other: Any) -> Any:
+        def jaccard_distance(self, other: Any) -> Operators:
             return self.op('<%>', return_type=Float)(other)
 
 

pgvector/sqlalchemy/halfvec.py

@@ -1,6 +1,6 @@
 from sqlalchemy.dialects.postgresql.base import ischema_names
 from sqlalchemy.types import UserDefinedType, Float, String
-from sqlalchemy import Dialect
+from sqlalchemy import Dialect, Operators
 from typing import Any
 from .. import HalfVector
 
@@ -36,16 +36,16 @@ def process(value):
         return process
 
     class comparator_factory(UserDefinedType.Comparator):
-        def l2_distance(self, other: Any) -> Any:
+        def l2_distance(self, other: Any) -> Operators:
             return self.op('<->', return_type=Float)(other)
 
-        def max_inner_product(self, other: Any) -> Any:
+        def max_inner_product(self, other: Any) -> Operators:
             return self.op('<#>', return_type=Float)(other)
 
-        def cosine_distance(self, other: Any) -> Any:
+        def cosine_distance(self, other: Any) -> Operators:
             return self.op('<=>', return_type=Float)(other)
 
-        def l1_distance(self, other: Any) -> Any:
+        def l1_distance(self, other: Any) -> Operators:
             return self.op('<+>', return_type=Float)(other)
 
 

pgvector/sqlalchemy/sparsevec.py

@@ -1,6 +1,6 @@
 from sqlalchemy.dialects.postgresql.base import ischema_names
 from sqlalchemy.types import UserDefinedType, Float, String
-from sqlalchemy import Dialect
+from sqlalchemy import Dialect, Operators
 from typing import Any
 from .. import SparseVector
 
@@ -36,16 +36,16 @@ def process(value):
         return process
 
     class comparator_factory(UserDefinedType.Comparator):
-        def l2_distance(self, other: Any) -> Any:
+        def l2_distance(self, other: Any) -> Operators:
             return self.op('<->', return_type=Float)(other)
 
-        def max_inner_product(self, other: Any) -> Any:
+        def max_inner_product(self, other: Any) -> Operators:
             return self.op('<#>', return_type=Float)(other)
 
-        def cosine_distance(self, other: Any) -> Any:
+        def cosine_distance(self, other: Any) -> Operators:
             return self.op('<=>', return_type=Float)(other)
 
-        def l1_distance(self, other: Any) -> Any:
+        def l1_distance(self, other: Any) -> Operators:
             return self.op('<+>', return_type=Float)(other)
 
 

pgvector/sqlalchemy/vector.py

@@ -1,6 +1,6 @@
 from sqlalchemy.dialects.postgresql.base import ischema_names
 from sqlalchemy.types import UserDefinedType, Float, String
-from sqlalchemy import Dialect
+from sqlalchemy import Dialect, Operators
 from typing import Any
 from .. import Vector
 
@@ -36,16 +36,16 @@ def process(value):
         return process
 
     class comparator_factory(UserDefinedType.Comparator):
-        def l2_distance(self, other: Any) -> Any:
+        def l2_distance(self, other: Any) -> Operators:
             return self.op('<->', return_type=Float)(other)
 
-        def max_inner_product(self, other: Any) -> Any:
+        def max_inner_product(self, other: Any) -> Operators:
             return self.op('<#>', return_type=Float)(other)
 
-        def cosine_distance(self, other: Any) -> Any:
+        def cosine_distance(self, other: Any) -> Operators:
             return self.op('<=>', return_type=Float)(other)
 
-        def l1_distance(self, other: Any) -> Any:
+        def l1_distance(self, other: Any) -> Operators:
             return self.op('<+>', return_type=Float)(other)