DeviceModel_V1.scala

sealed trait DeviceModel

object DeviceModel {
  implicit val deviceModelDataSource = new DataSource[DeviceModel, Unit] {
    override def read(parameters: Unit)(implicit spark: SparkSession) =
      AnnotatedDataFrame[DeviceModel](
        spark
          .createDataFrame(Seq(
            (0, "model_0"),
            (1, "model_1")
          ))
          .toDF("device_model_id", "model_name")
      )
  }
}