Made MyLabeledPoint into a Java Bean

apache · Nov 2, 2014 · 6fddc1c · 6fddc1c
1 parent 20630bc
commit 6fddc1c
Showing 1 changed file with 6 additions and 1 deletion.
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
@@ -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] {