This is an automated email from the ASF dual-hosted git repository. Amar3tto pushed a commit to branch fix-spanner-flatmap in repository https://gitbox.apache.org/repos/asf/beam.git
commit ba412b0ad7d64bc0e83b8eb10ca92e2e0e716c4d Author: Vitaly Terentyev <[email protected]> AuthorDate: Tue Oct 20 14:03:35 2026 +0400 Use with_output_types --- .../apache_beam/io/gcp/experimental/spannerio_read_perf_test.py | 4 +++- .../apache_beam/io/gcp/experimental/spannerio_write_perf_test.py | 4 +++- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/sdks/python/apache_beam/io/gcp/experimental/spannerio_read_perf_test.py b/sdks/python/apache_beam/io/gcp/experimental/spannerio_read_perf_test.py index 0fb97594d82..89502eadd3a 100644 --- a/sdks/python/apache_beam/io/gcp/experimental/spannerio_read_perf_test.py +++ b/sdks/python/apache_beam/io/gcp/experimental/spannerio_read_perf_test.py @@ -57,6 +57,7 @@ from apache_beam import FlatMap from apache_beam import Map from apache_beam import ParDo from apache_beam.io import Read +from apache_beam.io.gcp.experimental.spannerio import _Mutator from apache_beam.io.gcp.experimental.spannerio import ReadFromSpanner from apache_beam.io.gcp.experimental.spannerio import WriteToSpanner from apache_beam.testing.load_tests.load_test import LoadTest @@ -133,7 +134,8 @@ class SpannerReadPerfTest(LoadTest): | 'Produce rows' >> Read( SyntheticSource(self.parse_synthetic_source_options())) | 'Format' >> Map(format_record) - | 'Make mutations' >> FlatMap(make_insert_mutations) + | 'Make mutations' >> FlatMap( + make_insert_mutations).with_output_types(_Mutator) | 'Write to Spanner' >> WriteToSpanner( project_id=self.project, instance_id=self.spanner_instance, diff --git a/sdks/python/apache_beam/io/gcp/experimental/spannerio_write_perf_test.py b/sdks/python/apache_beam/io/gcp/experimental/spannerio_write_perf_test.py index 06be99fc633..6ec1a60e999 100644 --- a/sdks/python/apache_beam/io/gcp/experimental/spannerio_write_perf_test.py +++ b/sdks/python/apache_beam/io/gcp/experimental/spannerio_write_perf_test.py @@ -58,6 +58,7 @@ from apache_beam import FlatMap from apache_beam import Map from apache_beam import ParDo from apache_beam.io import Read +from apache_beam.io.gcp.experimental.spannerio import _Mutator from apache_beam.io.gcp.experimental.spannerio import WriteToSpanner from apache_beam.testing.load_tests.load_test import LoadTest from apache_beam.testing.load_tests.load_test_metrics_utils import CountMessages @@ -127,7 +128,8 @@ class SpannerWritePerfTest(LoadTest): SyntheticSource(self.parse_synthetic_source_options())) | 'Count messages' >> ParDo(CountMessages(self.metrics_namespace)) | 'Format' >> Map(format_record) - | 'Make mutations' >> FlatMap(make_insert_mutations) + | 'Make mutations' >> FlatMap( + make_insert_mutations).with_output_types(_Mutator) | 'Measure time' >> ParDo(MeasureTime(self.metrics_namespace)) | 'Write to Spanner' >> WriteToSpanner( project_id=self.project,
