Hyukjin Kwon created SPARK-41771:
------------------------------------
Summary: __getitem__ does not work with Column.isin
Key: SPARK-41771
URL: https://issues.apache.org/jira/browse/SPARK-41771
Project: Spark
Issue Type: Sub-task
Components: Connect
Affects Versions: 3.4.0
Reporter: Hyukjin Kwon
{code}
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
{code}
--
This message was sent by Atlassian Jira
(v8.20.10#820010)
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
