Fix invalid-type-form errors

Author: rok

python/pyproject.toml

@@ -106,7 +106,6 @@ rules.invalid-argument-type = "ignore"
 rules.invalid-assignment = "ignore"
 rules.invalid-context-manager = "ignore"
 rules.invalid-return-type = "ignore"
-rules.invalid-type-form = "ignore"
 rules.no-matching-overload = "ignore"
 rules.non-subscriptable = "ignore"
 rules.not-iterable = "ignore"

python/stubs/__lib_pxi/array.pyi

@@ -1,3 +1,4 @@
+import builtins
 import datetime as dt
 import sys
 
@@ -1990,7 +1991,7 @@ class Array(_PandasConvertible[pd.Series], Generic[_Scalar_co]):
     @overload
     def __getitem__(self, key: int) -> _Scalar_co: ...
     @overload
-    def __getitem__(self, key: slice) -> Self: ...
+    def __getitem__(self, key: builtins.slice) -> Self: ...
     def __getitem__(self, key):
         """
         Slice or return value at given index

python/stubs/__lib_pxi/io.pyi

@@ -1,3 +1,4 @@
+import builtins
 import sys
 
 from collections.abc import Callable
@@ -578,7 +579,7 @@ class Buffer(_Weakrefable):
     @property
     def parent(self) -> Buffer | None: ...
     @overload
-    def __getitem__(self, key: slice) -> Self: ...
+    def __getitem__(self, key: builtins.slice) -> Self: ...
     @overload
     def __getitem__(self, key: int) -> int: ...
     def slice(self, offset: int = 0, length: int | None = None) -> Self:

python/stubs/__lib_pxi/table.pyi

@@ -1,3 +1,4 @@
+import builtins
 import datetime as dt
 import sys
 
@@ -294,7 +295,7 @@ class ChunkedArray(_PandasConvertible[pd.Series], Generic[_Scalar_co]):
         """
     def __sizeof__(self) -> int: ...
     @overload
-    def __getitem__(self, key: slice) -> Self: ...
+    def __getitem__(self, key: builtins.slice) -> Self: ...
     @overload
     def __getitem__(self, key: int) -> _Scalar_co: ...
     def __getitem__(self, key):

python/stubs/compute.pyi

@@ -93,6 +93,7 @@ from . import lib
 
 _P = ParamSpec("_P")
 _R = TypeVar("_R")
+_CallableType = Callable[_P, _R]
 
 def field(*name_or_index: str | tuple[str, ...] | int) -> Expression:
     """Reference a column of the dataset.
@@ -156,7 +157,7 @@ def scalar(value: bool | float | str) -> Expression:
         An Expression representing the scalar value
     """
 
-def _clone_signature(f: Callable[_P, _R]) -> Callable[_P, _R]: ...
+def _clone_signature(f: _CallableType) -> _CallableType: ...
 
 # ============= compute functions =============
 _DataTypeT = TypeVar("_DataTypeT", bound=lib.DataType)