def __init__(self, env_fns): DummyVecEnv.__init__(self, env_fns) num_agents = getattr_unwrapped(self.envs[0], 'num_agents') VecMultiEnv.__init__(self, self.num_envs, num_agents, self.observation_space, self.action_space) self.buf_rews = np.zeros((self.num_envs, self.num_agents), dtype=np.float32)
def __init__(self, env_fns): DummyVecEnv.__init__(self, env_fns) self.buf_dones = np.zeros((self.num_envs,), dtype=np.ndarray) self.buf_rews = np.zeros((self.num_envs,), dtype=np.ndarray)