A few years back, the Step Functions team was soliciting input, and the Python thing was something that came up as a suggestion. It's hard, yes, but it should be possible to "Starlark" this and tell users that if you stick to this syntax, you can write Python and compile it down to native StepFunction syntax.
Having said that, they have slightly improved the StepFunctions by adopting JSONata syntax.
Having said that, they have slightly improved the StepFunctions by adopting JSONata syntax.