PySC2: Import of refs/pull/17/head

Move UI rendering to main thread for OSX compatibility PiperOrigin-RevId: 164995195
google-deepmind · Aug 11, 2017 · 509f607 · 509f607
1 parent 43a23ed
commit 509f607
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/pysc2/bin/agent.py b/pysc2/bin/agent.py
@@ -90,12 +90,13 @@ def _main(unused_argv):
  agent_cls = getattr(importlib.import_module(agent_module), agent_name)
 
  threads = []
- for i in range(FLAGS.parallel):
- t = threading.Thread(target=run_thread, args=(
- agent_cls, FLAGS.map, FLAGS.render and i == 0))
+ for _ in range(FLAGS.parallel - 1):
+ t = threading.Thread(target=run_thread, args=(agent_cls, FLAGS.map, False))
  threads.append(t)
  t.start()
 
+ run_thread(agent_cls, FLAGS.map, FLAGS.render)
+
  for t in threads:
  t.join()