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()]))