update scenes

Project/Assets/ML-Agents/Examples/Hallway/Scripts/HallwayCollabAgent.cs

 using UnityEngine;
-using Unity.MLAgents.Extensions.Teams;
+using Unity.MLAgents.Extensions.MultiAgent;
 using Unity.MLAgents.Actuators;
 using Unity.MLAgents.Sensors;
 
         base.Initialize();
         if (isSpotter)
         {
-            var teamManager = new HallwayTeamManager();
+            var teamManager = new BaseMultiAgentGroup();
             teamManager.RegisterAgent(this);
             teamManager.RegisterAgent(teammate);
         }

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

 using System.Collections;
 using System.Collections.Generic;
-using Unity.MLAgents.Extensions.Teams;
+using Unity.MLAgents.Extensions.MultiAgent;
 using UnityEngine;
 
 public class PushBlockEnvController : MonoBehaviour
 
     private int m_NumberOfRemainingBlocks;
 
-    private PushBlockTeamManager m_TeamManager;
+    private PushBlockAgentGroup m_AgentGroup;
 
     void Start()
     {
             item.Rb = item.T.GetComponent<Rigidbody>();
         }
         // Initialize TeamManager
-        m_TeamManager = new PushBlockTeamManager(this);
-        m_TeamManager.SetTeamMaxStep(MaxEnvironmentSteps);
+        m_AgentGroup = new PushBlockAgentGroup(this);
+        m_AgentGroup.SetGroupMaxStep(MaxEnvironmentSteps);
-            m_TeamManager.RegisterAgent(item.Agent);
+            m_AgentGroup.RegisterAgent(item.Agent);
         }
 
         ResetScene();
         if (done)
         {
             //Reset assets
-            m_TeamManager.EndTeamEpisode();
+            m_AgentGroup.EndGroupEpisode();
         }
     }
 

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

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

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

 fileFormatVersion: 2
-guid: f6ff089e647ef48ffbb569d14d170662
+guid: 41293044e9687478d8ca2d7df07ddad4
 MonoImporter:
   externalObjects: {}
   serializedVersion: 2

com.unity.ml-agents.extensions/Runtime/MultiAgent/BaseMultiAgentGroup.cs

             Reset();
         }
 
+        public virtual void OnGroupEpisodeBegin()
+        {
+
+        }
+
+            OnGroupEpisodeBegin();
         }
     }
 }

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/Hallway/Scripts/HallwayTeamManager.cs.meta

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

Project/Assets/ML-Agents/Examples/Hallway/Scripts/HallwayTeamManager.cs

-using System.Collections.Generic;
-using Unity.MLAgents;
-using Unity.MLAgents.Extensions.Teams;
-using Unity.MLAgents.Sensors;
-
-public class HallwayTeamManager : BaseTeamManager
-{
-    List<Agent> m_AgentList = new List<Agent> { };
-
-    public override void RegisterAgent(Agent agent)
-    {
-        m_AgentList.Add(agent);
-    }
-
-    // public override void OnAgentDone(Agent agent, Agent.DoneReason doneReason, List<ISensor> sensors)
-    // {
-
-    // }
-
-    // public override void AddTeamReward(float reward)
-    // {
-
-    // }
-}

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

-using System.Collections.Generic;
-using Unity.MLAgents;
-using Unity.MLAgents.Extensions.Teams;
-using Unity.MLAgents.Sensors;
-
-public class PushBlockTeamManager : BaseTeamManager
-{
-
-    PushBlockEnvController m_envController;
-
-    public PushBlockTeamManager(PushBlockEnvController envController)
-    {
-        m_envController = envController;
-    }
-
-    public override void OnTeamEpisodeBegin()
-    {
-        m_envController.ResetScene();
-    }
-}