Add missing params to Python Bigtable MutationsBatcher (#31791)

apache · Jul 10, 2024 · 018bcdf · 018bcdf
1 parent c9adc8e
commit 018bcdf
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/sdks/python/apache_beam/io/gcp/bigtableio.py b/sdks/python/apache_beam/io/gcp/bigtableio.py
@@ -141,7 +141,10 @@ def start_bundle(self):
  self.beam_options['instance_id'],
  self.beam_options['table_id'])
  self.batcher = MutationsBatcher(
- self.table, batch_completed_callback=self.write_mutate_metrics)
+ self.table,
+ batch_completed_callback=self.write_mutate_metrics,
+ flush_count=FLUSH_COUNT,
+ max_row_bytes=MAX_ROW_BYTES)
 
  def process(self, row):
  self.written.inc()