fix(python): Fix DataFrame.n_chunks return type (#5650)

pola-rs · Nov 28, 2022 · 908a4c7 · 908a4c7
1 parent 8497f95
commit 908a4c7
Showing 1 changed file with 1 addition and 5 deletions.
diff --git 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.