address comments

com.unity.ml-agents.extensions/Runtime/Teams/BaseTeamManager.cs

 
         public BaseTeamManager()
         {
-            Academy.Instance.TeamManagerStep += _ManagerStep;
+            Academy.Instance.PostAgentAct += _ManagerStep;
-            Academy.Instance.TeamManagerStep -= _ManagerStep;
+            Academy.Instance.PostAgentAct -= _ManagerStep;
         }
 
         void _ManagerStep()

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

         // This will mark the Agent as Done if it has reached its maxSteps.
         internal event Action AgentIncrementStep;
 
-        internal event Action TeamManagerStep;
+        internal event Action PostAgentAct;
 
         /// <summary>
         /// Signals to all of the <see cref="Agent"/>s that their step is about to begin.
                     AgentAct?.Invoke();
                 }
 
-                TeamManagerStep?.Invoke();
+                PostAgentAct?.Invoke();
             }
         }
 

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

             m_CumulativeReward += increment;
         }
 
-        public void SetTeamReward(float reward)
+        internal void SetTeamReward(float reward)
         {
 #if DEBUG
             Utilities.DebugCheckNanAndInfinity(reward, nameof(reward), nameof(SetTeamReward));
 
-        public void AddTeamReward(float increment)
+        internal void AddTeamReward(float increment)
         {
 #if DEBUG
             Utilities.DebugCheckNanAndInfinity(increment, nameof(increment), nameof(AddTeamReward));

com.unity.ml-agents/Runtime/Communicator/GrpcExtensions.cs

             var agentInfoProto = new AgentInfoProto
             {
                 Reward = ai.reward,
+                TeamReward = ai.teamReward,
                 MaxStepReached = ai.maxStepReached,
                 Done = ai.done,
                 Id = ai.episodeId,