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