add team reward

Project/Assets/ML-Agents/Examples/Hallway/TFModels/HallwayCollab.onnx.meta

   fileIDToRecycleName:
     11400000: main obj
     11400002: model data
-    2186277476908879412: ImportLogs
   externalObjects: {}
   userData: 
   assetBundleName: 

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

 
 public class PushBlockTeamManager : BaseTeamManager
 {
-    List<Agent> m_AgentList = new List<Agent> { };
+    Dictionary<Agent, bool> m_AgentDoneState = new Dictionary<Agent, bool> { };
-        m_AgentList.Add(agent);
+        m_AgentDoneState[agent] = false;
+    }
+
+    public override void OnAgentDone(Agent agent, Agent.DoneReason doneReason, List<ISensor> sensors)
+    {
+        m_AgentDoneState[agent] = true;
+        if (!m_AgentDoneState.ContainsValue(false))
+        {
+            foreach (var doneAgent in m_AgentDoneState.Keys)
+            {
+                doneAgent.SendDoneToTrainer();
+            }
+        }
+    }
+
+    public void ResetAgent(Agent agent)
+    {
+        if (!m_AgentDoneState.ContainsKey(agent))
+        {
+            RegisterAgent(agent);
+        }
+        else
+        {
+            m_AgentDoneState[agent] = false;
+        }
+    }
+
+    public override void AddTeamReward(float reward)
+    {
+        foreach (var agent in m_AgentDoneState.Keys)
+        {
+            agent.AddReward(reward);
+        }
     }
 }

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

         // col.gameObject.SetActive(false);
 
         //Give Agent Rewards
-        foreach (var item in AgentsList)
-        {
-            if (item.Agent.gameObject.activeInHierarchy)
-            {
-                print($"{item.Agent.name} scored");
-                item.Agent.AddReward(score);
-            }
-        }
+        // foreach (var item in AgentsList)
+        // {
+        //     if (item.Agent.gameObject.activeInHierarchy)
+        //     {
+        //         print($"{item.Agent.name} scored");
+        //         item.Agent.AddReward(score);
+        //     }
+        // }
+
+        m_TeamManager.AddTeamReward(score);
 
         // Swap ground material for a bit to indicate we scored.
         StartCoroutine(GoalScoredSwapGroundMaterial(m_PushBlockSettings.goalScoredMaterial, 0.5f));
     public void ZombieTouchedBlock()
     {
         //Give Agent Rewards
-        foreach (var item in AgentsList)
-        {
-            item.Agent.AddReward(-1);
-        }
+        // foreach (var item in AgentsList)
+        // {
+        //     item.Agent.AddReward(-1);
+        // }
+        m_TeamManager.AddTeamReward(-1);
         // Swap ground material for a bit to indicate we scored.
         StartCoroutine(GoalScoredSwapGroundMaterial(m_PushBlockSettings.failMaterial, 0.5f));
         ResetScene();
             item.Rb.velocity = Vector3.zero;
             item.Rb.angularVelocity = Vector3.zero;
             item.Agent.gameObject.SetActive(true);
+            m_TeamManager.ResetAgent(item.Agent);
         }
 
         //Reset Blocks

com.unity.ml-agents/Runtime/Agent.cs

 
         public void SendDoneToTrainer()
         {
+            m_Info.reward = m_Reward;
             // We request a decision so Python knows the Agent is done immediately
             m_Brain?.RequestDecision(m_Info, sensors);
             ResetSensors();