private def getSchemaDataTypeTransformationMap()

in scala-spark-sdk/src/main/scala/software/amazon/sagemaker/featurestore/sparksdk/validators/InputDataSchemaValidator.scala [181:194]

• 13 lines of code
• 1 McCabe index (conditional complexity)

  private def getSchemaDataTypeTransformationMap(
      validatorMap: Map[String, String => Column],
      featureDefinitions: List[FeatureDefinition],
      eventTimeFeatureName: String
  ): Map[String, String => Column] = {
    val eventTimeFeatureType =
      featureDefinitions.find(feature => feature.featureName().equals(eventTimeFeatureName)).get.featureTypeAsString()
    var dataTypeTransformationMap = validatorMap
    dataTypeTransformationMap += (eventTimeFeatureName -> ((featureName: String) =>
      col(featureName).cast(TYPE_MAP(eventTimeFeatureType))
    ))

    dataTypeTransformationMap
  }