diff --git a/pyarrow-stubs/_dataset.pyi b/pyarrow-stubs/_dataset.pyi index 6103a98..8475a97 100644 --- a/pyarrow-stubs/_dataset.pyi +++ b/pyarrow-stubs/_dataset.pyi @@ -351,6 +351,12 @@ class DirectoryPartitioning(KeyValuePartitioning): schema: lib.Schema | None = None, segment_encoding: Literal["uri", "none"] = "uri", ) -> PartitioningFactory: ... + def __init__( + self, + schema: lib.Schema, + dictionaries: dict[str, lib.Array] | None = None, + segment_encoding: Literal["uri", "none"] = "uri", + ) -> None: ... class HivePartitioning(KeyValuePartitioning): def __init__( diff --git a/pyarrow-stubs/dataset.pyi b/pyarrow-stubs/dataset.pyi index 4317b50..6bc0b4d 100644 --- a/pyarrow-stubs/dataset.pyi +++ b/pyarrow-stubs/dataset.pyi @@ -111,6 +111,10 @@ __all__ = [ _DatasetFormat: TypeAlias = Literal["parquet", "ipc", "arrow", "feather", "csv"] +@overload +def partitioning( + schema: Schema, +) -> Partitioning: ... @overload def partitioning( schema: Schema,