[SC-76491] Minor refactor

Author: Meng Tong <meng.tong@databricks.com> GitOrigin-RevId: 329ef04133858b76c512dbdd372dad2cee5fc6e5
delta-io · May 5, 2021 · e0f4cb5 · e0f4cb5
1 parent 52f55ca
commit e0f4cb5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/core/src/main/scala/org/apache/spark/sql/delta/GeneratedColumn.scala b/core/src/main/scala/org/apache/spark/sql/delta/GeneratedColumn.scala
@@ -40,7 +40,7 @@ import org.apache.spark.sql.connector.expressions.{BucketTransform, Transform}
 import org.apache.spark.sql.execution.{QueryExecution, SQLExecution}
 import org.apache.spark.sql.execution.datasources.HadoopFsRelation
 import org.apache.spark.sql.execution.streaming.IncrementalExecution
-import org.apache.spark.sql.types.{DataType, DateType, IntegerType, Metadata => FieldMetadata, MetadataBuilder, StringType, StructField, StructType, TimestampType}
+import org.apache.spark.sql.types.{DataType, DateType, DoubleType, FloatType, IntegerType, Metadata => FieldMetadata, MetadataBuilder, StringType, StructField, StructType, TimestampType}
 
 /**
  * Provide utility methods to implement Generated Columns for Delta. Users can use the following