update enjoy_split.py for issue #1

DKuan · Dec 4, 2019 · b495560 · b495560
1 parent ac5170c
commit b495560
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/enjoy_split.py b/enjoy_split.py
@@ -20,7 +20,7 @@ def make_env(scenario_name, arglist, benchmark=False):
         env = MultiAgentEnv(world, scenario.reset_world, scenario.reward, scenario.observation)
     return env
 
-def get_trainers(env, obs_shape_n, action_size, arglist):
+def get_trainers(env, arglist):
     trainers_cur = []
     trainers_tar = []
     optimizers = []
@@ -44,7 +44,7 @@ def enjoy(arglist):
 
     """ init the agents """
     obs_shape_n = [env.observation_space[i].shape for i in range(env.n)]
-    actors_tar = get_trainers(env, obs_shape_n, arglist.action_size, arglist)
+    actors_tar = get_trainers(env, arglist)
 
     """ interact with the env """
     obs_n = env.reset()