diff --git a/sql/core/src/test/scala/org/apache/spark/sql/UserDefinedTypeSuite.scala b/sql/core/src/test/scala/org/apache/spark/sql/UserDefinedTypeSuite.scala index 092b8fe12be2c..666235e57f812 100644 --- a/sql/core/src/test/scala/org/apache/spark/sql/UserDefinedTypeSuite.scala +++ b/sql/core/src/test/scala/org/apache/spark/sql/UserDefinedTypeSuite.scala @@ -17,6 +17,8 @@ package org.apache.spark.sql +import scala.beans.{BeanInfo, BeanProperty} + import org.apache.spark.rdd.RDD import org.apache.spark.sql.catalyst.annotation.SQLUserDefinedType import org.apache.spark.sql.catalyst.types.UserDefinedType @@ -31,7 +33,10 @@ private[sql] class MyDenseVector(val data: Array[Double]) extends Serializable { } } -private[sql] case class MyLabeledPoint(label: Double, features: MyDenseVector) +@BeanInfo +private[sql] case class MyLabeledPoint( + @BeanProperty label: Double, + @BeanProperty features: MyDenseVector) private[sql] class MyDenseVectorUDT extends UserDefinedType[MyDenseVector] {