def get_env(): env = gym.make('nesgym/NekketsuSoccerPK-v0') env = nesgym.wrap_nes_env(env) expt_dir = '/tmp/soccer/' env = wrappers.Monitor(env, os.path.join(expt_dir, "gym"), force=True) return env