Improved type hints [skip ci]

Author: ankane

pgvector/django/bit.py

@@ -22,8 +22,12 @@ def db_type(self, connection: Any) -> str:
             return 'bit'
         return 'bit(%d)' % self.length
 
-    def formfield(self, **kwargs: Any) -> forms.Field:  # type: ignore
-        return super().formfield(form_class=BitFormField, **kwargs)
+    def formfield(self, form_class: Any = None, choices_form_class: Any = None, **kwargs: Any) -> forms.Field:
+        return super().formfield(
+            form_class=BitFormField if form_class is None else form_class,
+            choices_form_class=choices_form_class,
+            **kwargs
+        )
 
 
 class BitFormField(forms.CharField):

pgvector/django/halfvec.py

@@ -41,8 +41,12 @@ def get_prep_value(self, value: Any) -> str | None:
     def value_to_string(self, obj: Any) -> str | None:
         return self.get_prep_value(self.value_from_object(obj))
 
-    def formfield(self, **kwargs) -> forms.Field:  # type: ignore
-        return super().formfield(form_class=HalfVectorFormField, **kwargs)
+    def formfield(self, form_class: Any = None, choices_form_class: Any = None, **kwargs: Any) -> forms.Field:
+        return super().formfield(
+            form_class=HalfVectorFormField if form_class is None else form_class,
+            choices_form_class=choices_form_class,
+            **kwargs
+        )
 
 
 class HalfVectorWidget(forms.TextInput):

pgvector/django/sparsevec.py

@@ -36,8 +36,12 @@ def get_prep_value(self, value: Any) -> str | None:
     def value_to_string(self, obj: Any) -> str | None:
         return self.get_prep_value(self.value_from_object(obj))
 
-    def formfield(self, **kwargs: Any) -> forms.Field:  # type: ignore
-        return super().formfield(form_class=SparseVectorFormField, **kwargs)
+    def formfield(self, form_class: Any = None, choices_form_class: Any = None, **kwargs: Any) -> forms.Field:
+        return super().formfield(
+            form_class=SparseVectorFormField if form_class is None else form_class,
+            choices_form_class=choices_form_class,
+            **kwargs
+        )
 
 
 class SparseVectorWidget(forms.TextInput):

pgvector/django/vector.py

@@ -49,8 +49,12 @@ def run_validators(self, value: Any) -> None:
             value = value.tolist()
         super().run_validators(value)
 
-    def formfield(self, **kwargs: Any) -> forms.Field:  # type: ignore
-        return super().formfield(form_class=VectorFormField, **kwargs)
+    def formfield(self, form_class: Any = None, choices_form_class: Any = None, **kwargs: Any) -> forms.Field:
+        return super().formfield(
+            form_class=VectorFormField if form_class is None else form_class,
+            choices_form_class=choices_form_class,
+            **kwargs
+        )
 
 
 class VectorWidget(forms.TextInput):