lint

Author: rok

python/pyarrow-stubs/__init__.pyi

@@ -15,13 +15,8 @@
 # specific language governing permissions and limitations
 # under the License.
 
-# ruff: noqa: F401, I001, E402
-__version__: str
-
 import pyarrow.lib as _lib
 
-_gc_enabled: bool
-
 from pyarrow.lib import (
     BuildInfo,
     RuntimeInfo,
@@ -39,13 +34,6 @@ from pyarrow.lib import (
     set_io_thread_count,
 )
 
-
-def show_versions() -> None: ...
-def show_info() -> None: ...
-def _module_is_available(module: str) -> bool: ...
-def _filesystem_is_available(fs: str) -> bool: ...
-
-
 from pyarrow.lib import (
     null,
     bool_,
@@ -251,7 +239,12 @@ from pyarrow.lib import (
 )
 
 # Buffers, allocation
-from pyarrow.lib import DeviceAllocationType, Device, MemoryManager, default_cpu_memory_manager
+from pyarrow.lib import (
+    DeviceAllocationType,
+    Device,
+    MemoryManager,
+    default_cpu_memory_manager
+)
 
 from pyarrow.lib import (
     Buffer,
@@ -351,6 +344,17 @@ from pyarrow.ipc import (
     RecordBatchStreamWriter,
 )
 
+
+__version__: str
+_gc_enabled: bool
+
+
+def show_versions() -> None: ...
+def show_info() -> None: ...
+def _module_is_available(module: str) -> bool: ...
+def _filesystem_is_available(fs: str) -> bool: ...
+
+
 # ----------------------------------------------------------------------
 # Returning absolute path to the pyarrow include directory (if bundled, e.g. in
 # wheels)

python/pyarrow-stubs/_compute.pyi

@@ -15,12 +15,15 @@
 # specific language governing permissions and limitations
 # under the License.
 
-from typing import (
-    Any,
+from collections.abc import (
     Callable,
     Iterable,
-    Literal,
     Sequence,
+)
+
+from typing import (
+    Any,
+    Literal,
     TypeAlias,
     TypedDict,
     overload,
@@ -189,8 +192,9 @@ class ExtractRegexSpanOptions(FunctionOptions):
 
 
 class FilterOptions(FunctionOptions):
-    def __init__(
-        self, null_selection_behavior: Literal["drop", "emit_null"] = "drop") -> None: ...
+    def __init__(self,
+                 null_selection_behavior: Literal["drop",
+                                                  "emit_null"] = "drop") -> None: ...
 
 
 class IndexOptions(FunctionOptions):
@@ -398,12 +402,16 @@ class SetLookupOptions(FunctionOptions):
 
 
 class SliceOptions(FunctionOptions):
-    def __init__(self, start: int, stop: int | None = None, step: int = 1) -> None: ...
+    def __init__(
+        self, start: int, stop: int | None = None, step: int = 1) -> None: ...
 
 
 class SortOptions(FunctionOptions):
     def __init__(
-        self, sort_keys: Sequence[tuple[str, _Order]], *, null_placement: _Placement = "at_end"
+        self,
+        sort_keys: Sequence[tuple[str, _Order]],
+        *,
+        null_placement: _Placement = "at_end"
     ) -> None: ...
 
 
@@ -424,15 +432,18 @@ class StrftimeOptions(FunctionOptions):
 
 
 class StrptimeOptions(FunctionOptions):
-    def __init__(
-        self, format: str, unit: Literal["s", "ms", "us", "ns"], error_is_null: bool = False
-    ) -> None: ...
+    def __init__(self,
+                 format: str,
+                 unit: Literal["s",
+                               "ms",
+                               "us",
+                               "ns"],
+                 error_is_null: bool = False) -> None: ...
 
 
 class StructFieldOptions(FunctionOptions):
-    def __init__(
-        self, indices: list[str] | list[bytes] | list[int] | Expression | bytes | str | int
-    ) -> None: ...
+    def __init__(self, indices: list[str] | list[bytes] |
+                 list[int] | Expression | bytes | str | int) -> None: ...
 
 
 class TakeOptions(FunctionOptions):
@@ -497,8 +508,9 @@ def list_functions() -> list[str]: ...
 
 
 def call_tabular_function(
-    function_name: str, args: Iterable | None = None, func_registry: FunctionRegistry | None = None
-) -> lib.RecordBatchReader: ...
+    function_name: str,
+    args: Iterable | None = None,
+    func_registry: FunctionRegistry | None = None) -> lib.RecordBatchReader: ...
 
 
 class _FunctionDoc(TypedDict):
@@ -569,10 +581,10 @@ class Expression(lib._Weakrefable):
     def __sub__(self, other) -> Expression: ...
     def __eq__(self, value: object) -> Expression: ...  # type: ignore[override]
     def __ne__(self, value: object) -> Expression: ...  # type: ignore[override]
-    def __gt__(self, value: object) -> Expression: ...  # type: ignore[override]
-    def __lt__(self, value: object) -> Expression: ...  # type: ignore[override]
-    def __ge__(self, value: object) -> Expression: ...  # type: ignore[override]
-    def __le__(self, value: object) -> Expression: ...  # type: ignore[override]
+    def __gt__(self, value: object) -> Expression: ...
+    def __lt__(self, value: object) -> Expression: ...
+    def __ge__(self, value: object) -> Expression: ...
+    def __le__(self, value: object) -> Expression: ...
     def __truediv__(self, other) -> Expression: ...
     def is_valid(self) -> bool: ...
     def is_null(self, nan_is_null: bool = False) -> Expression: ...

python/pyarrow-stubs/_csv.pyi

@@ -15,8 +15,9 @@
 # specific language governing permissions and limitations
 # under the License.
 
+from collections.abc import Callable
 from dataclasses import dataclass, field
-from typing import IO, Any, Callable, Literal
+from typing import IO, Any, Literal
 
 from _typeshed import StrPath
 
@@ -25,7 +26,7 @@ from . import lib
 
 @dataclass(kw_only=True)
 class ReadOptions(lib._Weakrefable):
-    use_threads: bool = field(default=True, kw_only=False)
+    use_threads: bool = field(default=True, kw_only=False)  # noqa: Y015
     block_size: int | None = None
     skip_rows: int = 0
     skip_rows_after_names: int = 0
@@ -38,7 +39,7 @@ class ReadOptions(lib._Weakrefable):
 @dataclass(kw_only=True)
 class ParseOptions(lib._Weakrefable):
 
-    delimiter: str = field(default=",", kw_only=False)
+    delimiter: str = field(default=",", kw_only=False)  # noqa: Y015
     quote_char: str | Literal[False] = '"'
     double_quote: bool = True
     escape_char: str | Literal[False] = False
@@ -52,7 +53,7 @@ class ParseOptions(lib._Weakrefable):
 @dataclass(kw_only=True)
 class ConvertOptions(lib._Weakrefable):
 
-    check_utf8: bool = field(default=True, kw_only=False)
+    check_utf8: bool = field(default=True, kw_only=False)  # noqa: Y015
     column_types: lib.Schema | dict | None = None
     null_values: list[str] | None = None
     true_values: list[str] | None = None
@@ -72,7 +73,7 @@ class ConvertOptions(lib._Weakrefable):
 @dataclass(kw_only=True)
 class WriteOptions(lib._Weakrefable):
 
-    include_header: bool = field(default=True, kw_only=False)
+    include_header: bool = field(default=True, kw_only=False)  # noqa: Y015
     batch_size: int = 1024
     delimiter: str = ","
     quoting_style: Literal["needed", "all_valid", "none"] = "needed"

python/pyarrow-stubs/_dataset.pyi

@@ -21,16 +21,14 @@ if sys.version_info >= (3, 11):
     from typing import Self
 else:
     from typing_extensions import Self
+from collections.abc import Callable, Iterator
 from typing import (
     IO,
     Any,
-    Callable,
     Generic,
-    Iterator,
     Literal,
     NamedTuple,
     TypeVar,
-    overload,
 )
 
 from _typeshed import StrPath

python/pyarrow-stubs/_dataset_parquet.pyi

@@ -15,8 +15,9 @@
 # specific language governing permissions and limitations
 # under the License.
 
+from collections.abc import Iterable
 from dataclasses import dataclass
-from typing import IO, Any, Iterable, TypedDict
+from typing import IO, Any, TypedDict
 
 from _typeshed import StrPath
 
@@ -112,8 +113,10 @@ class ParquetFileFragment(FileFragment):
 class ParquetReadOptions(_Weakrefable):
 
     def __init__(
-        self, dictionary_columns: list[str] | None, coerce_int96_timestamp_unit: str | None = None
-    ) -> None: ...
+        self,
+        dictionary_columns: list[str] | None,
+        coerce_int96_timestamp_unit: str | None = None) -> None: ...
+
     @property
     def coerce_int96_timestamp_unit(self) -> str: ...
     @coerce_int96_timestamp_unit.setter

python/pyarrow-stubs/_dataset_parquet_encryption.pyi

@@ -17,7 +17,8 @@
 
 from ._dataset_parquet import ParquetFileWriteOptions, ParquetFragmentScanOptions
 from ._parquet import FileDecryptionProperties
-from ._parquet_encryption import CryptoFactory, EncryptionConfiguration, KmsConnectionConfig
+from ._parquet_encryption import (
+    CryptoFactory, EncryptionConfiguration, KmsConnectionConfig)
 from .lib import _Weakrefable