def build_env(l: frozenlake.Lake): return frozenlake.FrozenLakeWithEscapingEnv(l, hole_retention_probability=0.99)
def build_env(lake: frozenlake.Lake): # return frozenlake.FrozenLakeEnv(lake, infinite_time=False) return frozenlake.FrozenLakeWithEscapingEnv( lake, hole_retention_probability=0.99)