From 93120e034725717608939f600c9da020de7b7671 Mon Sep 17 00:00:00 2001 From: Sven Mika Date: Wed, 26 Aug 2020 14:16:08 +0200 Subject: [PATCH] Unity3D API Fixes (recent changes in Unity's MLAgents API caused errors on RLlib side). (#10285) --- rllib/env/unity3d_env.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/rllib/env/unity3d_env.py b/rllib/env/unity3d_env.py index 61ad14348..4153a7b52 100644 --- a/rllib/env/unity3d_env.py +++ b/rllib/env/unity3d_env.py @@ -119,7 +119,7 @@ class Unity3DEnv(MultiAgentEnv): # Set only the required actions (from the DecisionSteps) in Unity3D. all_agents = [] - for behavior_name in self.unity_env.get_behavior_names(): + for behavior_name in self.unity_env.behavior_specs: for agent_id in self.unity_env.get_steps(behavior_name)[ 0].agent_id_to_index.keys(): key = behavior_name + "_{}".format(agent_id) @@ -166,7 +166,7 @@ class Unity3DEnv(MultiAgentEnv): obs = {} rewards = {} infos = {} - for behavior_name in self.unity_env.get_behavior_names(): + for behavior_name in self.unity_env.behavior_specs: decision_steps, terminal_steps = self.unity_env.get_steps( behavior_name) # Important: Only update those sub-envs that are currently