Skip to content

Commit 9ca514f

Browse files
fix: Do BigQuery indirect writes
Co-authored-by: Thomas Chow <[email protected]>
1 parent ce41459 commit 9ca514f

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

cloud_gcp/src/main/scala/ai/chronon/integrations/cloud_gcp/GcpFormatProvider.scala

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,10 @@ case class GcpFormatProvider(sparkSession: SparkSession) extends FormatProvider
4545
assert(scala.Option(tableId.getDataset).isDefined, s"dataset required for ${table}")
4646

4747
val sparkOptions: Map[String, String] = Map(
48-
"writeMethod" -> "direct",
48+
"temporaryGcsBucket" -> sparkSession.conf.get("spark.chronon.table.gcs.temporary_gcs_bucket"),
49+
"writeMethod" -> "indirect", // writeMethod direct does not output partitioned tables. keep as indirect.
50+
"materializationProject" -> tableId.getProject,
51+
"materializationDataset" -> tableId.getDataset,
4952
"createDisposition" -> JobInfo.CreateDisposition.CREATE_NEVER.name
5053
)
5154

0 commit comments

Comments
 (0)