itholic commented on code in PR #40525:
URL: https://github.com/apache/spark/pull/40525#discussion_r1159211030
##########
python/pyspark/pandas/spark/accessors.py:
##########
@@ -64,7 +68,7 @@ def column(self) -> Column:
"""
return self._data._internal.spark_column_for(self._data._column_label)
- def transform(self, func: Callable[[Column], Column]) -> IndexOpsLike:
+ def transform(self, func: Callable[[Column], Union[Column,
ConnectColumn]]) -> IndexOpsLike:
Review Comment:
ditto, let me address in subsequent PR since it brings another hundreds of
mypy failures? I plan to create several subsequent tickets after completing the
initial PR, so I'd want to focus on setting the test base first here in this PR
and encourage community involvement for the remaining errands.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
