Source code for openrl.envs.connect_env.connect3_env

from typing import Tuple

from openrl.envs.connect_env.base_connect_env import BaseConnectEnv


class Connect3Env(BaseConnectEnv):
    def _get_board_size(self) -> Tuple[int, int]:
        return 3, 3

    def _get_num2win(self) -> int:
        return 3


if __name__ == "__main__":
    env = Connect3Env(env_name="connect3")
    obs, info = env.reset()
    obs, reward, done, _, info = env.step(1, is_enemy=True)
    env.close()