mirror of
https://github.com/Farama-Foundation/Gymnasium.git
synced 2025-08-20 05:52:03 +00:00
168f25b976
Under Python3, user inputs are in string format and needs be converted into bytes before passing to pachi-py. The returned __repr__ from pachi-py for the boards are bytes and needs be converted into string for line breaks to work. Also fixed raw_input and print functions in play_go under examples/scripts with six package.
37 lines
925 B
Python
Executable File
37 lines
925 B
Python
Executable File
#!/usr/bin/env python
|
|
from six.moves import input as raw_input
|
|
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()
|