Merge pull request #547 from mozilla/issue546

Fixed #546
This commit is contained in:
Kelly Davis 2017-04-20 12:48:55 +02:00 committed by GitHub
commit 70da866e03
1 changed files with 1 additions and 0 deletions

View File

@ -66,6 +66,7 @@ class DataSet(object):
self._coord = coord
batch_threads = [Thread(target=self._populate_batch_queue, args=(session,)) for i in range(self._thread_count)]
for batch_thread in batch_threads:
self._coord.register_thread(batch_thread)
batch_thread.daemon = True
batch_thread.start()
return batch_threads