Fix deterministic policy evaluation

rlworkgroup · Jul 3, 2020 · f148dc6 · f148dc6
1 parent ea2b0ee
commit f148dc6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/garage/sampler/utils.py b/src/garage/sampler/utils.py
@@ -61,7 +61,7 @@ def rollout(env,
     while path_length < (max_path_length or np.inf):
         o = env.observation_space.flatten(o)
         a, agent_info = agent.get_action(o)
-        if deterministic and 'mean' in agent_infos:
+        if deterministic and 'mean' in agent_info:
             a = agent_info['mean']
         next_o, r, d, env_info = env.step(a)
         observations.append(o)