Add annotation for DataFrame.transform

zero323 · Jul 18, 2019 · 4a1da21 · 4a1da21
1 parent 979690a
commit 4a1da21
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/third_party/3/pyspark/sql/dataframe.pyi b/third_party/3/pyspark/sql/dataframe.pyi
@@ -145,6 +145,7 @@ class DataFrame:
     def drop(self, *cols: ColumnOrName) -> DataFrame: ...
     def toDF(self, *cols: ColumnOrName) -> DataFrame: ...
     def toPandas(self) -> pandas.core.frame.DataFrame: ...
+    def transform(self, func: Callable[[DataFrame], DataFrame]) -> DataFrame: ...
     @overload
     def groupby(self, *cols: ColumnOrName) -> DataFrame: ...
     @overload