fix tests

Unity-Technologies · Feb 11, 2021 · 492fd17 · 492fd17
1 parent 4e4ecad
commit 492fd17
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/ml-agents/mlagents/trainers/tests/simple_test_envs.py b/ml-agents/mlagents/trainers/tests/simple_test_envs.py
@@ -267,7 +267,9 @@ def _make_batched_step(
         m_group_id = np.array([0], dtype=np.int32)
         m_group_reward = np.array([group_reward], dtype=np.float32)
         action_mask = self._generate_mask()
-        decision_step = DecisionSteps(m_vector_obs, m_reward, m_agent_id, action_mask)
+        decision_step = DecisionSteps(
+            m_vector_obs, m_reward, m_agent_id, action_mask, m_group_id, m_group_reward
+        )
         terminal_step = TerminalSteps.empty(self.behavior_spec)
         if done:
             self.final_rewards[name].append(self.rewards[name])