Hi @Robert_Crowe thanks for posting this. Lines 34- 36 are particularly interesting:
flags.DEFINE_enum('model_framework', 'keras',
                  ['keras', 'flax_experimental', 'tfdf_experimental'],
                  'The modeling framework.')

I’m not familiar with how flags influence TFX - does specifying tfdf_experimental signal something special to either Google AI platform or the VertexAI platform when the pipeline is executed there?