Perhaps instead of initializing state and having to defeat the logger this code should be connecting to the agent and requesting state. It's a bit late to change this now but maybe for RL 6.0