feat: Use write api for automatic data uploads

TrevorBergeron · TrevorBergeron · commit 7f6481cd6923 · 2025-12-17T18:49:44.000Z
diff --git a/bigframes/session/bq_caching_executor.py b/bigframes/session/bq_caching_executor.py
@@ -594,7 +594,7 @@ def _upload_local_data(self, local_table: local_data.ManagedArrowTable):
         # Might be better as a queue and a worker thread
         with self._upload_lock:
             if local_table not in self.cache._uploaded_local_data:
-                uploaded = self.loader.load_data(
+                uploaded = self.loader.write_data(
                     local_table, bigframes.core.guid.generate_guid()
                 )
                 self.cache.cache_remote_replacement(local_table, uploaded)

Original file line number	Diff line number	Diff line change
`@@ -594,7 +594,7 @@ def _upload_local_data(self, local_table: local_data.ManagedArrowTable):`
`594`	`594`	`# Might be better as a queue and a worker thread`
`595`	`595`	`with self._upload_lock:`
`596`	`596`	`if local_table not in self.cache._uploaded_local_data:`
`597`		`- uploaded = self.loader.load_data(`
	`597`	`+ uploaded = self.loader.write_data(`
`598`	`598`	`local_table, bigframes.core.guid.generate_guid()`
`599`	`599`	`)`
`600`	`600`	`self.cache.cache_remote_replacement(local_table, uploaded)`