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])