Unity ML-Agents Limited Choice

[ilaydanil]

Hi,
In my game, my agent chooses 3 weapons out of multiple ones. And it starts its journey. This choosing weapon thing only happens once at the beginning of the game and it never happens again. How can I make the agent call this function and never call it again later?

I used OnActionReceived() and it chooses 3 weapons with no problem but I can not stop it after the first selection.

[MrWetsnow]

You can mask out those actions in a CollectDiscreteActionMasks() function. See:

Unity-Technologies/ml-agents/blob/main/docs/Learning-Environment-Design-Agents.md#masking-discrete-actions