rohdesamuel commented on a change in pull request #11838:
URL: https://github.com/apache/beam/pull/11838#discussion_r437784983
##########
File path: sdks/python/apache_beam/testing/test_stream.py
##########
@@ -291,10 +291,10 @@ def expand(self, pbegin):
assert isinstance(pbegin, pvalue.PBegin)
self.pipeline = pbegin.pipeline
if not self.output_tags:
- self.output_tags = set([None])
+ self.output_tags = {None}
Review comment:
This is a little harder to implement, mainly because the TestStream
retrieves its output_tags from the keys of the PTransform payload holding it.
This means that output_tags = None and output_tags = {None} look the same to
the PTransform payload outputs as a map with a single key being None. When a
TestStream is reconstructed, even if the original output_tags was unset, it
will be constructed with output_tags = {None}.
I think the best we can do is to treat {None} and None the same way.
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
