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

• Type: Sub-task
• Status: Resolved
• Priority: Major
• Resolution: Fixed
• Affects Version/s: 3.4.0
• Fix Version/s: 3.4.0
• Component/s: Connect
• Assignee: Ruifeng Zheng
• Reporter: Hyukjin Kwon
• Created: 29/Dec/22 12:09
• Updated: 30/Dec/22 08:11
• Resolved: 30/Dec/22 08:11

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

Issue Links

• links to: [Github] Pull Request #39300 (zhengruifeng)
• links to: [Github] Pull Request #39300 (zhengruifeng)
• links to: [Github] Pull Request #39300 (zhengruifeng)