Description
>>> foo = pandas_udf(lambda x: x, 'v int', PandasUDFType.GROUPED_MAP) Traceback (most recent call last): File "<stdin>", line 1, in <module> NameError: name 'PandasUDFType' is not defined
When explicitly imported it works fine:
>>> from pyspark.sql.functions import PandasUDFType >>> foo = pandas_udf(lambda x: x, 'v int', PandasUDFType.GROUPED_MAP)
We just need to make sure it's included in _all_/