Github user HyukjinKwon commented on a diff in the pull request:
https://github.com/apache/spark/pull/22568#discussion_r220879251
--- Diff: python/pyspark/sql/tests.py ---
@@ -5525,32 +5525,69 @@ def data(self):
.withColumn("v", explode(col('vs'))).drop('vs')
def test_supported_types(self):
- from pyspark.sql.functions import pandas_udf, PandasUDFType,
array, col
- df = self.data.withColumn("arr", array(col("id")))
+ from pyspark.sql.functions import pandas_udf, PandasUDFType
+ from decimal import Decimal
--- End diff --
not a big deal but
```
from decimal import Decimal
from pyspark.sql.functions import pandas_udf, PandasUDFType
```
---
---------------------------------------------------------------------
To unsubscribe, e-mail: reviews-unsubscr...@spark.apache.org
For additional commands, e-mail: reviews-h...@spark.apache.org
