dchvn commented on a change in pull request #35236:
URL: https://github.com/apache/spark/pull/35236#discussion_r787324507
##
File path: python/pyspark/pandas/typedef/typehints.py
##
@@ -559,6 +559,9 @@ def infer_return_type(f: Callable) -> Union[SeriesType,
DataFrameType,
dchvn commented on a change in pull request #35236:
URL: https://github.com/apache/spark/pull/35236#discussion_r786426897
##
File path: python/pyspark/pandas/tests/test_typedef.py
##
@@ -56,6 +56,23 @@
class TypeHintTests(unittest.TestCase):
+def