Have you tried using a custom_config argument for Trainer?  Something like:
trainer = tfx.components.Trainer(
    ...
    custom_config={
        'tft_stats':transform.outputs['post_transform_stats']
        }
)