From 9056783c962cd1093eee3eca718bdbc5b42e554a Mon Sep 17 00:00:00 2001 From: Sourabh Bajaj Date: Wed, 19 Jul 2017 10:08:14 -0700 Subject: [PATCH] [BEAM-2636] Make sure we only override the correct class --- sdks/python/apache_beam/runners/dataflow/dataflow_runner.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/sdks/python/apache_beam/runners/dataflow/dataflow_runner.py b/sdks/python/apache_beam/runners/dataflow/dataflow_runner.py index 89c18d4c0f53..aec7d0042e25 100644 --- a/sdks/python/apache_beam/runners/dataflow/dataflow_runner.py +++ b/sdks/python/apache_beam/runners/dataflow/dataflow_runner.py @@ -418,6 +418,9 @@ def run_Flatten(self, transform_node): PropertyNames.OUTPUT_NAME: PropertyNames.OUT}]) def apply_WriteToBigQuery(self, transform, pcoll): + # Make sure this is the WriteToBigQuery class that we expected + if not isinstance(transform, beam.io.WriteToBigQuery): + return self.apply_PTransform(transform, pcoll) standard_options = pcoll.pipeline._options.view_as(StandardOptions) if standard_options.streaming: if (transform.write_disposition ==