Repository: beam Updated Branches: refs/heads/master eb0850ef8 -> 2e51bde5b
[BEAM-2636] Make sure we only override the correct class Project: http://git-wip-us.apache.org/repos/asf/beam/repo Commit: http://git-wip-us.apache.org/repos/asf/beam/commit/d128c3b3 Tree: http://git-wip-us.apache.org/repos/asf/beam/tree/d128c3b3 Diff: http://git-wip-us.apache.org/repos/asf/beam/diff/d128c3b3 Branch: refs/heads/master Commit: d128c3b378a58b0c2c31c2d30fd29e211e118324 Parents: eb0850e Author: Sourabh Bajaj <[email protected]> Authored: Wed Jul 19 10:08:14 2017 -0700 Committer: Ahmet Altay <[email protected]> Committed: Wed Jul 19 14:07:54 2017 -0700 ---------------------------------------------------------------------- sdks/python/apache_beam/runners/dataflow/dataflow_runner.py | 3 +++ 1 file changed, 3 insertions(+) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/beam/blob/d128c3b3/sdks/python/apache_beam/runners/dataflow/dataflow_runner.py ---------------------------------------------------------------------- diff --git a/sdks/python/apache_beam/runners/dataflow/dataflow_runner.py b/sdks/python/apache_beam/runners/dataflow/dataflow_runner.py index 89c18d4..aec7d00 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 @@ class DataflowRunner(PipelineRunner): 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 ==
