Fixed typechecking errors [skip ci]

Author: ankane

pgvector/django/bit.py

@@ -21,7 +21,7 @@ def db_type(self, connection):
             return 'bit'
         return 'bit(%d)' % self.length
 
-    def formfield(self, **kwargs):
+    def formfield(self, **kwargs):  # type: ignore
         return super().formfield(form_class=BitFormField, **kwargs)
 
 

pgvector/django/halfvec.py

@@ -40,7 +40,7 @@ def get_prep_value(self, value):
     def value_to_string(self, obj):
         return self.get_prep_value(self.value_from_object(obj))
 
-    def formfield(self, **kwargs):
+    def formfield(self, **kwargs):  # type: ignore
         return super().formfield(form_class=HalfVectorFormField, **kwargs)
 
 

pgvector/django/sparsevec.py

@@ -35,7 +35,7 @@ def get_prep_value(self, value):
     def value_to_string(self, obj):
         return self.get_prep_value(self.value_from_object(obj))
 
-    def formfield(self, **kwargs):
+    def formfield(self, **kwargs):  # type: ignore
         return super().formfield(form_class=SparseVectorFormField, **kwargs)
 
 

pgvector/django/vector.py

@@ -48,7 +48,7 @@ def run_validators(self, value):
             value = value.tolist()
         super().run_validators(value)
 
-    def formfield(self, **kwargs):
+    def formfield(self, **kwargs):  # type: ignore
         return super().formfield(form_class=VectorFormField, **kwargs)
 
 

pgvector/sqlalchemy/bit.py

@@ -35,4 +35,4 @@ def jaccard_distance(self, other):
 
 
 # for reflection
-ischema_names['bit'] = BIT
+ischema_names['bit'] = BIT  # type: ignore

pgvector/sqlalchemy/halfvec.py

@@ -25,7 +25,7 @@ def literal_processor(self, dialect):
         string_literal_processor = self._string._cached_literal_processor(dialect)
 
         def process(value):
-            return string_literal_processor(HalfVector._to_db(value, self.dim))
+            return string_literal_processor(HalfVector._to_db(value, self.dim))  # type: ignore
         return process
 
     def result_processor(self, dialect, coltype):
@@ -48,4 +48,4 @@ def l1_distance(self, other):
 
 
 # for reflection
-ischema_names['halfvec'] = HALFVEC
+ischema_names['halfvec'] = HALFVEC  # type: ignore

pgvector/sqlalchemy/sparsevec.py

@@ -25,7 +25,7 @@ def literal_processor(self, dialect):
         string_literal_processor = self._string._cached_literal_processor(dialect)
 
         def process(value):
-            return string_literal_processor(SparseVector._to_db(value, self.dim))
+            return string_literal_processor(SparseVector._to_db(value, self.dim))  # type: ignore
         return process
 
     def result_processor(self, dialect, coltype):
@@ -48,4 +48,4 @@ def l1_distance(self, other):
 
 
 # for reflection
-ischema_names['sparsevec'] = SPARSEVEC
+ischema_names['sparsevec'] = SPARSEVEC  # type: ignore

pgvector/sqlalchemy/vector.py

@@ -25,7 +25,7 @@ def literal_processor(self, dialect):
         string_literal_processor = self._string._cached_literal_processor(dialect)
 
         def process(value):
-            return string_literal_processor(Vector._to_db(value, self.dim))
+            return string_literal_processor(Vector._to_db(value, self.dim))  # type: ignore
         return process
 
     def result_processor(self, dialect, coltype):
@@ -48,4 +48,4 @@ def l1_distance(self, other):
 
 
 # for reflection
-ischema_names['vector'] = VECTOR
+ischema_names['vector'] = VECTOR  # type: ignore