def __init__(self, config, create_env, create_train_env, create_agent):
    self.config = config
    self.logger = TFLogger(log_dir=self.config['logdir'], hps=self.config)
    self._create_env = create_env
    self._create_train_env = create_train_env
    self._create_agent = create_agent
    env = self._create_env(self.config['n_envs'], seed=0, env_name=self.config['env_name'])
    self.n_actions = env.action_space.n
    self.obs_dim = env.reset()[0]['frame'].size()[1]
    del env
def __init__(self, config, create_env, create_train_env, create_agent):
    self.config = config
    self.logger = TFLogger(log_dir=self.config['logdir'], hps=self.config)
    self._create_env = create_env
    self._create_train_env = create_train_env
    self._create_agent = create_agent
    env = self._create_env(self.config['n_envs'], seed=0, env_name=self.config['env_name'])
    self.n_actions = env.action_space.n
    self.obs_dim = env.reset()[0]['frame'].size()[1]
    del env