from gym.envs.registration import register

register(
    id='Gomoku19x19-v0',
    entry_point='gym_gomoku.envs:GomokuEnv',
    kwargs={
        'player_color': 'black',
        'opponent': 'beginner', # beginner opponent policy has defend and strike rules
        'board_size': 19,
    },
    nondeterministic=True,
)

register(
    id='Gomoku9x9-v0',
    entry_point='gym_gomoku.envs:GomokuEnv',
    kwargs={
        'player_color': 'black',
        'opponent': 'beginner', # random policy is the simplest
        'board_size': 9,
    },
    nondeterministic=True,
)

register(
    id='Gomoku19x19duel-v0',
    entry_point='gym_gomoku.envs:GomokuEnv',
    kwargs={
        'player_color': 'black',
        'opponent': 'userdefine', # random policy is the simplest
        'board_size': 19,
    },
    nondeterministic=True,
)

register(
    id='Gomoku9x9duel-v0',
    entry_point='gym_gomoku.envs:GomokuEnv',
    kwargs={
        'player_color': 'black',
        'opponent': 'userdefine', # random policy is the simplest
        'board_size': 9,
    },
    nondeterministic=True,
)


