[BEAM-14217] Figure out how type hints should work

• Type: Sub-task
• Status: Open
• Priority: P2
• Resolution: Unresolved
• Component/s: sdk-py-core
• Assignee: Unassigned
• Reporter: Ryan Thompson
• Created: 31/Mar/22 14:50
• Updated: 04/Jun/22 23:31

Description

RunInference main interface could have many different types of input and ouput types.

 

It probably should have generice typehints of some sort, but the code that infers them will need to be specified.

 

I haven't yet figured out how to do this, but I'm convinced that getting really specific type hints will help our users find errors.

 

We have

@input output types unknown

RunInference(PTransform)
  has RunInferenceDoFun (also input output types unknown)
    calls composedRunInferenceFn (input/output types known)

we need to get correct type hints into the RunInference PTransform and it's not apparent how to do that exactly since annotations won't be possible.

 

I think we'll need some correct values to put into the with_output_type chain in the ptransform.