python overload vstack (#1997)

pola-rs · Dec 6, 2021 · dacef40 · dacef40
1 parent 807209e
commit dacef40
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/py-polars/polars/internals/frame.py b/py-polars/polars/internals/frame.py
@@ -2493,6 +2493,18 @@ def hstack(
         else:
             return wrap_df(self._df.hstack([s.inner() for s in columns]))
 
+    @tp.overload
+    def vstack(self, df: "DataFrame", in_place: Literal[True]) -> None:
+        ...
+
+    @tp.overload
+    def vstack(self, df: "DataFrame", in_place: Literal[False] = ...) -> "DataFrame":
+        ...
+
+    @tp.overload
+    def vstack(self, df: "DataFrame", in_place: bool) -> Optional["DataFrame"]:
+        ...
+
     def vstack(self, df: "DataFrame", in_place: bool = False) -> Optional["DataFrame"]:
         """
         Grow this DataFrame vertically by stacking a DataFrame to it.