Skip to content

Commit a891110

Browse files
add back bq filter
Co-authored-by: Thomas Chow <[email protected]> Co-authored-by: Thomas Chow <[email protected]> Co-authored-by: Thomas Chow <[email protected]>
1 parent a17857a commit a891110

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

spark/src/main/scala/ai/chronon/spark/TableUtils.scala

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -330,9 +330,10 @@ class TableUtils(@transient val sparkSession: SparkSession) extends Serializable
330330
stats: Option[DfStats] = None,
331331
sortByCols: Seq[String] = Seq.empty): Unit = {
332332
// partitions to the last
333+
val dataPointer = DataPointer.from(tableName, sparkSession)
333334
val colOrder = df.columns.diff(partitionColumns) ++ partitionColumns
334335
val dfRearranged: DataFrame = df.select(colOrder.map {
335-
case c if c == partitionColumn =>
336+
case c if c == partitionColumn && dataPointer.writeFormat.map(_.toUpperCase).exists("BIGQUERY".equals) =>
336337
to_date(df.col(c), partitionFormat).as(partitionColumn)
337338
case c => df.col(c)
338339
}: _*)

0 commit comments

Comments
 (0)