Is there an idiom for using case logic to define paramters to tasks without duplicating the entire flow inside the case? Something like this?

with case(is_rapids(input), True):
    export_result = ExportsTask()(secret_task, 'somekey', stack)

with case(is_rapids(input), False):
    export_result = ExportsTask()(secret_task, 'anotherkey', stack)

MyTask()(input, export_result['training_job_queue'], export_result['training_job_definition'], PipelineArtifactsBucketTask()(secret_task, stack))