Преобразование кадра данных в набор данных с использованием искомого класса case

Question

Я написал следующий код, целью которого является преобразование кадра данных в набор данных с использованием класса наблюдения

def toDs[T](df: DataFrame): Dataset[T] = {
    df.as[T]
  } 

, затем case class DATA( name:String, age:Double, location:String)

Я получаю:

Unable to find encoder for type stored in a Dataset.  Primitive types (Int, String, etc) and Product types (case classes) are supported by importing spark.implicits._  Support for serializing other types will be added in future releases.
[error]     df.as[T]

Любая идея, как это исправить