20 lines
512 B
Python
20 lines
512 B
Python
import sys
|
|
from game import Game
|
|
from engine import GTPEngine
|
|
# import utils
|
|
import argparse
|
|
import time
|
|
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("--checkpoint_path", type=str, default="./checkpoints/")
|
|
args = parser.parse_args()
|
|
|
|
game = Game(checkpoint_path=args.checkpoint_path)
|
|
engine = GTPEngine(game_obj=game, name='tianshou', version=0)
|
|
|
|
while not engine.disconnect:
|
|
command = sys.stdin.readline()
|
|
result = engine.run_cmd(command)
|
|
sys.stdout.write(result)
|
|
sys.stdout.flush()
|