Add missing acceptable types to sql.DataFrame (#399)

See https://github.com/apache/spark/blob/branch-3.0/python/pyspark/sql/dataframe.py#L1369-L1371 for additional acceptable types. Co-authored-by: Maciej <zero323@users.noreply.github.com>
zero323 · Apr 28, 2020 · 5f10c18 · 5f10c18
1 parent 13f4958
commit 5f10c18
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/third_party/3/pyspark/sql/dataframe.pyi b/third_party/3/pyspark/sql/dataframe.pyi
@@ -128,7 +128,7 @@ class DataFrame:
     @overload
     def head(self, n: int) -> List[Row]: ...
     def first(self) -> Row: ...
-    def __getitem__(self, item: Union[int, str]) -> Column: ...
+    def __getitem__(self, item: Union[int, str, Column, List, Tuple]) -> Column: ...
     def __getattr__(self, name: str) -> Column: ...
     @overload
     def select(self, *cols: ColumnOrName) -> DataFrame: ...