Is there any way to modify on_step function during interaction with the environment?

[Felixvillas]

Hi~

Thanks for sharing this great job!

I'm trying to develop a LLM(Large Language Model)-based agent, which can interact with the environment by generating code. In my understanding, this agent may generate different codes at different times since the state of the environment may change. The different codes I am referring to here refer to modifying the on_step function during interaction with the environment.

Or, is there any other way to do this in python-sc2?