init xp_per_marine

chris-chris · Nov 18, 2017 · d7531d8 · d7531d8
1 parent a18195b
commit d7531d8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/a2c/a2c.py b/a2c/a2c.py
@@ -261,7 +261,7 @@ def __init__(self,
     self.base_act_mask = np.full((self.nenv, 2), 0, dtype=np.uint8)
     obs, rewards, dones, available_actions, army_counts, control_groups, selected, xy_per_marine = env.reset(
     )
-    self.xy_per_marine = [{} for _ in range(nenv)]
+    self.xy_per_marine = [{"0":[0,0], "1":[0,0]} for _ in range(nenv)]
     for env_num, data in enumerate(xy_per_marine):
       self.xy_per_marine[env_num] = data
     self.army_counts = army_counts