From 51948b3cd0de346385b1153e4b568a270486177b Mon Sep 17 00:00:00 2001 From: Heejong Lee Date: Fri, 10 Sep 2021 12:12:18 -0700 Subject: [PATCH] [BEAM-12805] Fix XLang CombinePerKey test by explicitly assigning the output type --- .../runners/portability/expansion_service_test.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/sdks/python/apache_beam/runners/portability/expansion_service_test.py b/sdks/python/apache_beam/runners/portability/expansion_service_test.py index 1c02f710c01a..98d2faa89c6b 100644 --- a/sdks/python/apache_beam/runners/portability/expansion_service_test.py +++ b/sdks/python/apache_beam/runners/portability/expansion_service_test.py @@ -185,9 +185,12 @@ def from_runner_api_parameter( @ptransform.PTransform.register_urn(TEST_COMPK_URN, None) class CombinePerKeyTransform(ptransform.PTransform): def expand(self, pcoll): - return pcoll \ - | beam.CombinePerKey(sum).with_output_types( - typing.Tuple[str, int]) + output = pcoll \ + | beam.CombinePerKey(sum) + # TODO: Use `with_output_types` instead of explicitly + # assigning to `.element_type` after fixing BEAM-12872 + output.element_type = beam.typehints.Tuple[str, int] + return output def to_runner_api_parameter(self, unused_context): return TEST_COMPK_URN, None