Repository: beam Updated Branches: refs/heads/release-2.1.0 c7c777d8b -> 04899e414
[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/ab21078b Tree: http://git-wip-us.apache.org/repos/asf/beam/tree/ab21078b Diff: http://git-wip-us.apache.org/repos/asf/beam/diff/ab21078b Branch: refs/heads/release-2.1.0 Commit: ab21078b0ea88f610d164245c379da8634ecd854 Parents: 1cf560b Author: Sourabh Bajaj <[email protected]> Authored: Wed Jul 19 10:08:14 2017 -0700 Committer: Sourabh Bajaj <[email protected]> Committed: Mon Jul 24 09:08:03 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/ab21078b/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 57bcc5e..4b91ca4 100644 --- a/sdks/python/apache_beam/runners/dataflow/dataflow_runner.py +++ b/sdks/python/apache_beam/runners/dataflow/dataflow_runner.py @@ -429,6 +429,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 ==
