Add TeamManager

Project/Assets/ML-Agents/Examples/CubeWars/Scripts/CubeWarArea.cs

     [HideInInspector]
     public LargeCubeAgent[] largeAgents;
     public float range;
+    private CubeWarsTeamManager m_SmallTeamManager;
+    private CubeWarsTeamManager m_LargeTeamManager;
+        m_SmallTeamManager = new CubeWarsTeamManager();
+        m_LargeTeamManager = new CubeWarsTeamManager();
+        foreach (var agent in smallAgents)
+        {
+            agent.SetTeamManager(m_SmallTeamManager);
+        }
+        foreach (var agent in largeAgents)
+        {
+            agent.SetTeamManager(m_LargeTeamManager);
+        }
     }
 
     public void AgentDied()

Project/Assets/ML-Agents/Examples/CubeWars/Scripts/CubeWarsTeamManager.cs

+using System.Collections.Generic;
+using Unity.MLAgents;
+using Unity.MLAgents.Extensions.Teams;
+using Unity.MLAgents.Sensors;
+
+public class CubeWarsTeamManager : BaseTeamManager
+{
+    List<Agent> m_AgentList = new List<Agent> { };
+
+
+    public override void RegisterAgent(Agent agent)
+    {
+        m_AgentList.Add(agent);
+    }
+}