Cucmumber datatable to spark dataframe

// Cucmumber datatable to spark dataframe
import scala.collection.convert.ImplicitConversions.`collection AsScalaIterable`
import io.cucumber.datatable.DataTable
import org.apache.spark.sql.DataFrame

def dataTableToDataframe(table: DataTable): DataFrame = {
  import sparkSession.implicits._
  val columns: Seq[String] = table.cells().head.toSeq
  val data = table.cells().drop(1).toSeq.map(r => r.toList)
  data.toDF().select(columns.indices.map(i => col("value")(i).alias(columns(i))): _*)
}