import sys. Fixes #55

tensorflow · Feb 5, 2018 · 14f64b1 · 14f64b1
1 parent 9905f5e
commit 14f64b1
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/selfplay_mcts.py b/selfplay_mcts.py
@@ -16,13 +16,15 @@
 import time
 import numpy as np
 import random
+import sys
 
 import coords
 import go
 from gtp_wrapper import MCTSPlayer
 
 SIMULTANEOUS_LEAVES = 8
 
+
 def play(network, readouts, resign_threshold, verbosity=0):
     ''' Plays out a self-play match, returning
     - the final position
@@ -60,7 +62,8 @@ def play(network, readouts, resign_threshold, verbosity=0):
             print(player.root.describe())
 
         if player.should_resign():
-            player.set_result(-1 * player.root.position.to_play, was_resign=True)
+            player.set_result(-1 * player.root.position.to_play,
+                              was_resign=True)
             break
         move = player.pick_move()
         player.play_move(move)