minor fix

jgocm · Feb 24, 2021 · 221688a · 221688a
1 parent 016d556
commit 221688a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/envs/agents/deepvss/deepvss_ma.py b/envs/agents/deepvss/deepvss_ma.py
@@ -74,7 +74,7 @@ def dict_to_str(dct):
 
         sys.stdout.flush()
 
-        n_robots = args.n_robots
+        n_robots = args.n_agents
         env = gym.make('VSSMA-v0', n_robots_blue=n_robots,
                        n_robots_yellow=n_robots)
         state_shape, action_shape = env.observation_space, env.action_space