reset when done

Project/Assets/ML-Agents/Examples/PushBlock/Scripts/PushBlockEnvController.cs

         {
             //Reset assets
             m_AgentGroup.EndGroupEpisode();
+            ResetScene();
         }
     }