hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-06 10:31:39 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix Tuple spaces in rollout.py (#5332)

Browse source 
Make sure that the initial action is also properly flattened.
This commit is contained in:
Jaroslaw Rzepecki 2019-07-31 19:38:49 +01:00 • committed by Eric Liang
parent e218e615df
commit b3c8091a35
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
python/ray/rllib/rollout.py
View file

@ -132,7 +132,7 @@ def rollout(agent, env_name, num_steps, out=None, no_render=True):
state_init = {p: m.get_initial_state() for p, m in policy_map.items()}
use_lstm = {p: len(s) > 0 for p, s in state_init.items()}
action_init = {
p: m.action_space.sample()
p: _flatten_action(m.action_space.sample())
for p, m in policy_map.items()
}
else: