From 018bcdf592ce9aa6ec6d09f86a5a5da728c54c78 Mon Sep 17 00:00:00 2001 From: Minh Son Nguyen Date: Wed, 10 Jul 2024 12:58:49 +0300 Subject: [PATCH] Add missing params to Python Bigtable MutationsBatcher (#31791) --- sdks/python/apache_beam/io/gcp/bigtableio.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/sdks/python/apache_beam/io/gcp/bigtableio.py b/sdks/python/apache_beam/io/gcp/bigtableio.py index 0f3944a791bdc..3f54e09ee3dd6 100644 --- 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()