from gym.envs.registration import registry, register, make, spec

register(
        id='AntMazeBottleneck-v0',
        entry_point='envs.antenv.ant_maze_bottleneck:AntMazeBottleneckEnv',
        max_episode_steps=500,
        reward_threshold=0.0,
    )

register(
        id='AntMazeBottleneck-eval-v0',
        entry_point='envs.antenv.ant_maze_bottleneck:AntMazeBottleneckEvalEnv',
        max_episode_steps=500,
        reward_threshold=0.0,
    )

register(
        id='Reacher3D-v0',
        entry_point='envs.fetchenv.create_fetch_env:create_fetch_env',
        kwargs={'env_name': 'Reacher3D-v0'},
        max_episode_steps=100
)

register(
        id='Pusher-v0',
        entry_point='envs.fetchenv.create_fetch_env:create_fetch_env',
        kwargs={'env_name': 'Pusher-v0'},
        max_episode_steps=100
)