9ddf959bb26d5d15d4a0c9ddd6821a262866214c,a2c/a2c.py,Runner,update_obs,#Runner#,231

Before Change

    self.group_id = [0 for _ in range(nenv)]

  def update_obs(self, obs): // (self.nenv, 32, 32, 2)
    obs = np.asarray(obs, dtype=np.int32).swapaxes(1, 2).swapaxes(2, 3)
    self.obs = np.roll(self.obs, shift=-1, axis=3)
    self.obs[:, :, :, -1:] = obs[:, :, :, :]
    // could not broadcast input array from shape (4,1,32,32) into shape (4,4,32)

After Change

  def update_obs(self, obs): // (self.nenv, 32, 32, 2)
    //obs = np.asarray(obs, dtype=np.int32).swapaxes(1, 2).swapaxes(2, 3)
    self.obs = np.roll(self.obs, shift=-3, axis=3)
    new_map = np.zeros((self.nenv, 32, 32, 3))
    new_map[:,:,:,-1] = obs[:,0,:,:]
    for env_num in range(self.nenv):
      marine0 = self.xy_per_marine[env_num]["0"]
      marine1 = self.xy_per_marine[env_num]["1"]

Instances