I know we could submit a python wheel job through UI, but is that possible to achieve it with airflow DatabricksSubmitRunOperator?

My understanding is that the DatabricksWorkflowTaskGroup is inheriting from a TaskGroup which doesn’t perform template rendering,
The branch name is defined in my variables json files, one for staging and one for production.
I’m not sure what’s the best option for me in this case.