This is an automated email from the ASF dual-hosted git repository.
tvalentyn pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/beam.git
The following commit(s) were added to refs/heads/master by this push:
new 428ec97e30c improve error message for mismatched pipelines (#24834)
428ec97e30c is described below
commit 428ec97e30cc6587c2ae0f81d3ba44a8a9b34f93
Author: ZhengLin Li <63448884+zhenglin...@users.noreply.github.com>
AuthorDate: Thu Jan 19 16:53:30 2023 -0800
improve error message for mismatched pipelines (#24834)
Co-authored-by: tvalentyn <tvalen...@users.noreply.github.com>
---
sdks/python/apache_beam/transforms/util.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/sdks/python/apache_beam/transforms/util.py
b/sdks/python/apache_beam/transforms/util.py
index 5f99d680546..868dbe1361b 100644
--- a/sdks/python/apache_beam/transforms/util.py
+++ b/sdks/python/apache_beam/transforms/util.py
@@ -234,7 +234,8 @@ class _CoGBKImpl(PTransform):
for pcoll in pcolls.values():
self._check_pcollection(pcoll)
if self.pipeline:
- assert pcoll.pipeline == self.pipeline
+ assert pcoll.pipeline == self.pipeline, (
+ 'All input PCollections must belong to the same pipeline.')
tags = list(pcolls.keys())
