Github user ueshin commented on a diff in the pull request: https://github.com/apache/spark/pull/20531#discussion_r166832153 --- Diff: python/pyspark/sql/tests.py --- @@ -4562,14 +4585,14 @@ def test_basic(self): self.assertPandasEqual(expected4.toPandas(), result4.toPandas()) def test_unsupported_types(self): - from pyspark.sql.types import ArrayType, DoubleType, MapType + from pyspark.sql.types import DoubleType, MapType from pyspark.sql.functions import pandas_udf, PandasUDFType with QuietTest(self.sc): with self.assertRaisesRegexp(NotImplementedError, 'not supported'): - @pandas_udf(ArrayType(DoubleType()), PandasUDFType.GROUPED_AGG) + @pandas_udf(ArrayType(ArrayType(TimestampType())), PandasUDFType.GROUPED_AGG) def mean_and_std_udf(v): --- End diff -- nit: should rename this?
--- --------------------------------------------------------------------- To unsubscribe, e-mail: reviews-unsubscr...@spark.apache.org For additional commands, e-mail: reviews-h...@spark.apache.org