def create_env(cls, flags, featurizer=None): f = featurizer or X.Concat([X.Text(), X.ValidMoves()]) print('loading env') start_time = time.time() env = gym.make(flags.env, room_shape=(flags.height, flags.width), partially_observable=flags.partial_observability, max_placement=flags.max_placement, featurizer=f, shuffle_wiki=flags.shuffle_wiki, time_penalty=flags.time_penalty) print('loaded env in {} seconds'.format(time.time() - start_time)) return env
def create_env(cls, flags, featurizer=None): return super().create_env(flags, featurizer=featurizer or X.Concat([X.Text(), X.ValidMoves(), X.RelativePosition()]))