Repository: beam Updated Branches: refs/heads/master 22ff898a3 -> 50fc63a9b
Avoid Runner API translation of pipelines with PDone. Project: http://git-wip-us.apache.org/repos/asf/beam/repo Commit: http://git-wip-us.apache.org/repos/asf/beam/commit/a41bbbb9 Tree: http://git-wip-us.apache.org/repos/asf/beam/tree/a41bbbb9 Diff: http://git-wip-us.apache.org/repos/asf/beam/diff/a41bbbb9 Branch: refs/heads/master Commit: a41bbbb9a8e58cafced10232ff56094a58fe17ce Parents: 22ff898 Author: Robert Bradshaw <[email protected]> Authored: Sat Apr 1 12:42:40 2017 -0700 Committer: Robert Bradshaw <[email protected]> Committed: Tue Apr 4 11:26:53 2017 -0700 ---------------------------------------------------------------------- sdks/python/apache_beam/pipeline.py | 5 +++++ 1 file changed, 5 insertions(+) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/beam/blob/a41bbbb9/sdks/python/apache_beam/pipeline.py ---------------------------------------------------------------------- diff --git a/sdks/python/apache_beam/pipeline.py b/sdks/python/apache_beam/pipeline.py index 3c416eb..8506b85 100644 --- a/sdks/python/apache_beam/pipeline.py +++ b/sdks/python/apache_beam/pipeline.py @@ -318,6 +318,11 @@ class Pipeline(object): pickler.loads(pickler.dumps(transform_node.transform)) except Exception: Visitor.ok = False + + def visit_value(self, value, _): + if isinstance(value, pvalue.PDone): + Visitor.ok = False + self.visit(Visitor()) return Visitor.ok
