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

Commit 816a88c

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 3df4220 commit 816a88c

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
@@ -140,7 +140,7 @@ class DataFrame(PandasMapOpsMixin, PandasConversionMixin):
140140
@overload
141141
def head(self, n: int) -> List[Row]: ...
142142
def first(self) -> Row: ...
143-
def __getitem__(self, item: Union[int, str]) -> Column: ...
143+
def __getitem__(self, item: Union[int, str, Column, List, Tuple]) -> Column: ...
144144
def __getattr__(self, name: str) -> Column: ...
145145
@overload
146146
def select(self, *cols: ColumnOrName) -> DataFrame: ...

0 commit comments

Comments
 (0)