update scene

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

 
     private int m_NumberOfRemainingBlocks;
 
-    private PushBlockAgentGroup m_AgentGroup;
+    private BaseMultiAgentGroup m_AgentGroup;
+
+    private int m_ResetTimer;
 
     void Start()
     {
             item.Rb = item.T.GetComponent<Rigidbody>();
         }
         // Initialize TeamManager
-        m_AgentGroup = new PushBlockAgentGroup(this);
-        m_AgentGroup.SetGroupMaxStep(MaxEnvironmentSteps);
+        m_AgentGroup = new BaseMultiAgentGroup();
         foreach (var item in AgentsList)
         {
             item.StartingPos = item.Agent.transform.position;
 
         ResetScene();
 
+    }
+
+    void FixedUpdate()
+    {
+        m_ResetTimer += 1;
+        if (m_ResetTimer >= MaxEnvironmentSteps && MaxEnvironmentSteps > 0)
+        {
+            m_AgentGroup.GroupEpisodeInterrupted();
+            ResetScene();
+        }
     }
 
     /// <summary>
 
     public void ResetScene()
     {
-        // m_ResetTimer = 0;
+        m_ResetTimer = 0;
 
         //Random platform rot
         var rotation = Random.Range(0, 4);

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

     PushBlockSettings m_PushBlockSettings;
 
     private int m_NumberOfRemainingBlocks;
-    private PushBlockAgentGroup m_TeamManager;
 
     void Start()
     {

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

-using System;
-using System.Collections.Generic;
-using Unity.MLAgents;
-using Unity.MLAgents.Extensions.MultiAgent;
-using Unity.MLAgents.Sensors;
-
-using System.Collections.Generic;
-using System.Collections.ObjectModel;
-using UnityEngine;
-using Unity.Barracuda;
-using Unity.MLAgents.Actuators;
-using Unity.MLAgents.Sensors;
-using Unity.MLAgents.Sensors.Reflection;
-using Unity.MLAgents.Demonstrations;
-using Unity.MLAgents.Policies;
-using UnityEngine.Serialization;
-
-public class PushBlockAgentGroup : BaseMultiAgentGroup
-{
-
-    PushBlockEnvController m_envController;
-    Action ResetScene;
-
-    public PushBlockAgentGroup(PushBlockEnvController envController)
-    {
-        ResetScene = envController.ResetScene;
-    }
-
-    public override void OnGroupEpisodeBegin()
-    {
-        ResetScene.Invoke();
-    }
-}

Project/Assets/ML-Agents/Examples/PushBlock/Scripts/PushBlockAgentGroup.cs.meta

-fileFormatVersion: 2
-guid: 41293044e9687478d8ca2d7df07ddad4
-MonoImporter:
-  externalObjects: {}
-  serializedVersion: 2
-  defaultReferences: []
-  executionOrder: 0
-  icon: {instanceID: 0}
-  userData: 
-  assetBundleName: 
-  assetBundleVariant: