From 6fb987ed1495f292441edb9775ef6415fc28210e Mon Sep 17 00:00:00 2001 From: Dipam Chakraborty <dipam@aicrowd.com> Date: Thu, 3 Jun 2021 02:18:24 +0000 Subject: [PATCH] Update agent.py --- agent.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/agent.py b/agent.py index dd60c43..2a1c391 100644 --- a/agent.py +++ b/agent.py @@ -84,7 +84,8 @@ if __name__ == '__main__': infos = [{} for _ in range(num_envs)] # Change this to your agent interface - num_actions = batched_env.envs[0].action_space.n + # num_actions = batched_env.envs[0].action_space.n + num_actions = 112 agent = RandomBatchedAgent(num_envs, num_actions) # The evaluation setup will automatically stop after the requisite number of rollouts -- GitLab