diff --git a/src/main/java/io/cdap/plugin/gcp/bigquery/sqlengine/BigQuerySparkDatasetProducer.java b/src/main/java/io/cdap/plugin/gcp/bigquery/sqlengine/BigQuerySparkDatasetProducer.java index b4de6b60f3..4eae3fbd25 100644 --- a/src/main/java/io/cdap/plugin/gcp/bigquery/sqlengine/BigQuerySparkDatasetProducer.java +++ b/src/main/java/io/cdap/plugin/gcp/bigquery/sqlengine/BigQuerySparkDatasetProducer.java @@ -123,13 +123,13 @@ private Dataset convertFieldTypes(Dataset ds) { } // Handle Int types - if (fieldSchema.getType() == Schema.Type.INT && fieldSchema.getLogicalType() == null) { + if (fieldSchema.getType() == Schema.Type.INT) { LOG.trace("Converting field {} to Integer", fieldName); ds = ds.withColumn(fieldName, ds.col(fieldName).cast(DataTypes.IntegerType)); } // Handle float types - if (fieldSchema.getType() == Schema.Type.FLOAT && fieldSchema.getLogicalType() == null) { + if (fieldSchema.getType() == Schema.Type.FLOAT) { LOG.trace("Converting field {} to Float", fieldName); ds = ds.withColumn(fieldName, ds.col(fieldName).cast(DataTypes.FloatType)); }