wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-09-09 11:32:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

Unity3D API Fixes (recent changes in Unity's MLAgents API caused errors on RLlib side). (#10285)

Browse Source
This commit is contained in:
Sven Mika
2020-08-26 14:16:08 +02:00
committed by GitHub
parent 4e9888ce2f
commit 93120e0347
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rllib/env/unity3d_env.py
Vendored
+2 -2
View File
@@ -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