Fixed typechecking [skip ci]

Author: ankane

pgvector/sqlalchemy/bit.py

@@ -28,13 +28,15 @@ def process(value: Any) -> Any:
         else:
             return super().bind_processor(dialect)
 
-    class comparator_factory(TypeEngine.Comparator):
+    class Comparator(TypeEngine.Comparator):
         def hamming_distance(self, other: object) -> Operators:
             return self.op('<~>', return_type=Float)(other)
 
         def jaccard_distance(self, other: object) -> Operators:
             return self.op('<%>', return_type=Float)(other)
 
+    comparator_factory = Comparator
+
 
 # for reflection
 ischema_names['bit'] = BIT  # type: ignore

pgvector/sqlalchemy/halfvec.py

@@ -35,7 +35,7 @@ def process(value: Any) -> HalfVector | None:
             return HalfVector._from_db(value)
         return process
 
-    class comparator_factory(TypeEngine.Comparator):
+    class Comparator(TypeEngine.Comparator):
         def l2_distance(self, other: object) -> Operators:
             return self.op('<->', return_type=Float)(other)
 
@@ -48,6 +48,8 @@ def cosine_distance(self, other: object) -> Operators:
         def l1_distance(self, other: object) -> Operators:
             return self.op('<+>', return_type=Float)(other)
 
+    comparator_factory = Comparator
+
 
 # for reflection
 ischema_names['halfvec'] = HALFVEC  # type: ignore

pgvector/sqlalchemy/sparsevec.py

@@ -35,7 +35,7 @@ def process(value: Any) -> SparseVector | None:
             return SparseVector._from_db(value)
         return process
 
-    class comparator_factory(TypeEngine.Comparator):
+    class Comparator(TypeEngine.Comparator):
         def l2_distance(self, other: object) -> Operators:
             return self.op('<->', return_type=Float)(other)
 
@@ -48,6 +48,8 @@ def cosine_distance(self, other: object) -> Operators:
         def l1_distance(self, other: object) -> Operators:
             return self.op('<+>', return_type=Float)(other)
 
+    comparator_factory = Comparator
+
 
 # for reflection
 ischema_names['sparsevec'] = SPARSEVEC  # type: ignore

pgvector/sqlalchemy/vector.py

@@ -35,7 +35,7 @@ def process(value: Any) -> Vector | None:
             return Vector._from_db(value)
         return process
 
-    class comparator_factory(TypeEngine.Comparator):
+    class Comparator(TypeEngine.Comparator):
         def l2_distance(self, other: object) -> Operators:
             return self.op('<->', return_type=Float)(other)
 
@@ -48,6 +48,8 @@ def cosine_distance(self, other: object) -> Operators:
         def l1_distance(self, other: object) -> Operators:
             return self.op('<+>', return_type=Float)(other)
 
+    comparator_factory = Comparator
+
 
 # for reflection
 ischema_names['vector'] = VECTOR  # type: ignore