diff --git a/py-polars/polars/internals/dataframe/frame.py b/py-polars/polars/internals/dataframe/frame.py index 55126bdeb198..396cfe6ad80e 100644 --- a/py-polars/polars/internals/dataframe/frame.py +++ b/py-polars/polars/internals/dataframe/frame.py @@ -5606,17 +5606,13 @@ def with_columns( ) @overload - def n_chunks(self, strategy: Literal["first"]) -> int: + def n_chunks(self, strategy: Literal["first"] = ...) -> int: ... @overload def n_chunks(self, strategy: Literal["all"]) -> list[int]: ... - @overload - def n_chunks(self, strategy: str = "first") -> int | list[int]: - ... - def n_chunks(self, strategy: str = "first") -> int | list[int]: """ Get number of chunks used by the ChunkedArrays of this DataFrame.