Improved type hints [skip ci]

Author: ankane

pgvector/django/bit.py

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

pgvector/django/halfvec.py

@@ -41,7 +41,7 @@ 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):  # type: ignore
+    def formfield(self, **kwargs) -> forms.Field:  # type: ignore
         return super().formfield(form_class=HalfVectorFormField, **kwargs)
 
 

pgvector/django/sparsevec.py

@@ -36,7 +36,7 @@ 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):  # type: ignore
+    def formfield(self, **kwargs: Any) -> forms.Field:  # type: ignore
         return super().formfield(form_class=SparseVectorFormField, **kwargs)
 
 

pgvector/django/vector.py

@@ -49,7 +49,7 @@ def run_validators(self, value: Any) -> None:
             value = value.tolist()
         super().run_validators(value)
 
-    def formfield(self, **kwargs: Any):  # type: ignore
+    def formfield(self, **kwargs: Any) -> forms.Field:  # type: ignore
         return super().formfield(form_class=VectorFormField, **kwargs)
 
 

pgvector/peewee/bit.py

@@ -5,7 +5,7 @@
 class FixedBitField(Field):
     field_type = 'bit'
 
-    def __init__(self, max_length: int | None = None, *args, **kwargs) -> None:
+    def __init__(self, max_length: int | None = None, *args: Any, **kwargs: Any) -> None:
         self.max_length = max_length
         super(FixedBitField, self).__init__(*args, **kwargs)
 

pgvector/peewee/halfvec.py

@@ -6,7 +6,7 @@
 class HalfVectorField(Field):
     field_type = 'halfvec'
 
-    def __init__(self, dimensions: int | None = None, *args, **kwargs) -> None:
+    def __init__(self, dimensions: int | None = None, *args: Any, **kwargs: Any) -> None:
         self.dimensions = dimensions
         super(HalfVectorField, self).__init__(*args, **kwargs)
 

pgvector/peewee/sparsevec.py

@@ -6,7 +6,7 @@
 class SparseVectorField(Field):
     field_type = 'sparsevec'
 
-    def __init__(self, dimensions: int | None = None, *args, **kwargs) -> None:
+    def __init__(self, dimensions: int | None = None, *args: Any, **kwargs: Any) -> None:
         self.dimensions = dimensions
         super(SparseVectorField, self).__init__(*args, **kwargs)
 

pgvector/peewee/vector.py

@@ -7,7 +7,7 @@
 class VectorField(Field):
     field_type = 'vector'
 
-    def __init__(self, dimensions: int | None = None, *args, **kwargs) -> None:
+    def __init__(self, dimensions: int | None = None, *args: Any, **kwargs: Any) -> None:
         self.dimensions = dimensions
         super(VectorField, self).__init__(*args, **kwargs)
 

pgvector/sqlalchemy/bit.py

@@ -11,7 +11,7 @@ def __init__(self, length: int | None = None) -> None:
         super(UserDefinedType, self).__init__()
         self.length = length
 
-    def get_col_spec(self, **kw) -> str:
+    def get_col_spec(self, **kw: Any) -> str:
         if self.length is None:
             return 'BIT'
         return 'BIT(%d)' % self.length

pgvector/sqlalchemy/halfvec.py

@@ -13,7 +13,7 @@ def __init__(self, dim: int | None = None) -> None:
         super(UserDefinedType, self).__init__()
         self.dim = dim
 
-    def get_col_spec(self, **kw) -> str:
+    def get_col_spec(self, **kw: Any) -> str:
         if self.dim is None:
             return 'HALFVEC'
         return 'HALFVEC(%d)' % self.dim