[SPARK-41771] __getitem__ does not work with Column.isin - ASF JIRA

Rank to Top

Rank to Bottom

Attach files

Attach Screenshot

Bulk Copy Attachments

Bulk Move Attachments

Voters

Watch issue

Watchers

Convert to Issue

Link

Clone

Labels

Update Comment Author

Replace String in Comment

Update Comment Visibility

Delete Comments

XML

Word

Printable

JSON

Details

Type: Sub-task
Status: Resolved
Priority: Major
Resolution: Fixed
Affects Version/s: 3.4.0
Fix Version/s: 3.4.0
Component/s: Connect
Labels:
None

Description

File "/.../spark/python/pyspark/sql/connect/column.py", line 364, in pyspark.sql.connect.column.Column.isin
Failed example:
    df[df.name.isin("Bob", "Mike")].collect()
Exception raised:
    Traceback (most recent call last):
      File "/.../miniconda3/envs/python3.9/lib/python3.9/doctest.py", line 1336, in __run
        exec(compile(example.source, filename, "single",
      File "<doctest pyspark.sql.connect.column.Column.isin[1]>", line 1, in <module>
        df[df.name.isin("Bob", "Mike")].collect()
      File "/.../workspace/forked/spark/python/pyspark/sql/connect/dataframe.py", line 888, in __getitem__
        return col(name)
      File "/.../workspace/forked/spark/python/pyspark/sql/connect/functions.py", line 161, in col
        return Column(ColumnReference(col))
      File "/.../spark/python/pyspark/sql/connect/expressions.py", line 322, in __init__
        assert isinstance(name, str)
    AssertionError
**********************************************************************
File "/.../spark/python/pyspark/sql/connect/column.py", line 366, in pyspark.sql.connect.column.Column.isin
Failed example:
    df[df.age.isin([1, 2, 3])].collect()
Exception raised:
    Traceback (most recent call last):
      File "/.../miniconda3/envs/python3.9/lib/python3.9/doctest.py", line 1336, in __run
        exec(compile(example.source, filename, "single",
      File "<doctest pyspark.sql.connect.column.Column.isin[2]>", line 1, in <module>
        df[df.age.isin([1, 2, 3])].collect()
      File "/.../spark/python/pyspark/sql/connect/dataframe.py", line 888, in __getitem__
        return col(name)
      File "/.../spark/python/pyspark/sql/connect/functions.py", line 161, in col
        return Column(ColumnReference(col))
      File "/.../spark/python/pyspark/sql/connect/expressions.py", line 322, in __init__
        assert isinstance(name, str)
    AssertionError