Typehint a doFn with multiple outputs in python

[Quentin Sommer]

Hi,

I can't seem to find in the documentation how to add type hints to a doFn
doing multiple tagged output. My function looks like this:

values = beam.FlatMap(
  parse_sessions.sessions_unnest
).with_outputs(
  OUTPU1,
  OUTPU2,
  OUTPU3
)

I'm looking for a way to have a correct typehint later when I use one of
the outputs like this:

values[OUTPUT1] | beam.Map(some_function)

Best,
Quentin