mirror of
https://github.com/Farama-Foundation/Gymnasium.git
synced 2025-08-20 05:52:03 +00:00
36 lines
882 B
Plaintext
36 lines
882 B
Plaintext
|
#!/usr/bin/env python
|
||
|
|
import argparse
|
||
|
|
import pachi_py
|
||
|
|
import gym
|
||
|
|
from gym import spaces, envs
|
||
|
|
from gym.envs.board_game import go
|
||
|
|
|
||
|
|
def main():
|
||
|
|
parser = argparse.ArgumentParser()
|
||
|
|
parser.add_argument('--raw_actions', action='store_true')
|
||
|
|
args = parser.parse_args()
|
||
|
|
|
||
|
|
env = envs.make('Go9x9-v0')
|
||
|
|
env.reset()
|
||
|
|
while True:
|
||
|
|
s = env._state
|
||
|
|
env._render()
|
||
|
|
|
||
|
|
colorstr = pachi_py.color_to_str(s.color)
|
||
|
|
if args.raw_actions:
|
||
|
|
a = int(raw_input('{} (raw)> '.format(colorstr)))
|
||
|
|
else:
|
||
|
|
coordstr = raw_input('{}> '.format(colorstr))
|
||
|
|
a = go.str_to_action(s.board, coordstr)
|
||
|
|
|
||
|
|
_, r, done, _ = env.step(a)
|
||
|
|
if done:
|
||
|
|
break
|
||
|
|
|
||
|
|
print
|
||
|
|
print 'You win!' if r > 0 else 'Opponent wins!'
|
||
|
|
print 'Final score:', env._state.board.official_score
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
main()
|