Skip to content
This repository was archived by the owner on Nov 22, 2022. It is now read-only.

Commit 5f10c18

Browse files
radeklatzero323
andcommitted
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 <[email protected]>
1 parent 13f4958 commit 5f10c18

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

third_party/3/pyspark/sql/dataframe.pyi

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -128,7 +128,7 @@ class DataFrame:
128128
@overload
129129
def head(self, n: int) -> List[Row]: ...
130130
def first(self) -> Row: ...
131-
def __getitem__(self, item: Union[int, str]) -> Column: ...
131+
def __getitem__(self, item: Union[int, str, Column, List, Tuple]) -> Column: ...
132132
def __getattr__(self, name: str) -> Column: ...
133133
@overload
134134
def select(self, *cols: ColumnOrName) -> DataFrame: ...

0 commit comments

Comments
 (0)