def save_env_fn(self, env_fn):
    fname = os.path.join(self.logdir, 'checkpoints/env_fn.pkl')
    U.save(fname, env_fn)
def save_env_fn(self, env_fn):
    fname = os.path.join(self.logdir, 'checkpoints/env_fn.pkl')
    U.save(fname, env_fn)