Restart protocol not working

[JIElite]

Environment:

• BWAPI: 4.1.2
• Torchcraft 1.0.2

I use SingleBattleEnv as my environment, and hard code env._reset() when it reach max_steps. like:

if steps >  max_steps:
    env._reset()

But, the restart protocol doesn't work! The game still running.