Fixed CI

Author: ankane

pgvector/sqlalchemy/bit.py

@@ -14,11 +14,9 @@ def process_bind_param(self, value: Any, dialect: Any) -> Any:
             return asyncpg.BitString(value)  # type: ignore
         return value
 
-    class Comparator(TypeDecorator.Comparator[Any]):
+    class comparator_factory(TypeDecorator.Comparator[Any]):
         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  # type: ignore

pgvector/sqlalchemy/halfvec.py

@@ -35,7 +35,7 @@ def process(value: Any) -> list[float] | None:
             return HalfVector._from_db(value)
         return process
 
-    class Comparator(TypeEngine.Comparator[Any]):
+    class comparator_factory(TypeEngine.Comparator[Any]):
         def l2_distance(self, other: object, /) -> Operators:
             return self.op('<->', return_type=Float)(other)
 
@@ -48,8 +48,6 @@ 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(TypeEngine.Comparator[Any]):
+    class comparator_factory(TypeEngine.Comparator[Any]):
         def l2_distance(self, other: object, /) -> Operators:
             return self.op('<->', return_type=Float)(other)
 
@@ -48,8 +48,6 @@ 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) -> list[float] | None:
             return Vector._from_db(value)
         return process
 
-    class Comparator(TypeEngine.Comparator[Any]):
+    class comparator_factory(TypeEngine.Comparator[Any]):
         def l2_distance(self, other: object, /) -> Operators:
             return self.op('<->', return_type=Float)(other)
 
@@ -48,8 +48,6 @@ 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