[BEAM-10833] Error inferring type of tuple output

• Type: Bug
• Status: Resolved
• Priority: P1
• Resolution: Fixed
• Fix Version/s: 2.25.0
• Component/s: sdk-py-core
• Assignee: Unassigned
• Reporter: Robert Bradshaw
• Created: 28/Aug/20 18:00
• Updated: 15/May/21 17:55
• Resolved: 15/May/21 17:55

Description

with beam.Pipeline() as p:

  class Foo(beam.DoFn):
    def process(self, element):
      k, v = element
      yield (k, {k:v})

  a = [('a',1), ('a',1), ('b',1)]

  input = p | beam.Create(a)
  print(input.element_type)  # ok

  pcoll = input | beam.ParDo(Foo())
  print(pcoll.element_type) # claims Tuple[int, Dict[Any, Any]]

# BOOM on run due to coder errors

Issue Links

• links to: GitHub Pull Request #12798