Run _TPUPollingThread as a daemon thread

PiperOrigin-RevId: 263609561
tensorflow · Aug 15, 2019 · 36e672c · 36e672c
1 parent 928ec43
commit 36e672c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tensorflow/python/tpu/preempted_hook.py b/tensorflow/python/tpu/preempted_hook.py
@@ -60,6 +60,7 @@ class _TPUPollingThread(threading.Thread):
   def __init__(self, cluster, session):
     super(_TPUPollingThread, self).__init__()
 
+    self.daemon = True
     self._running = True
     self._session_closed = False
     self._cluster = cluster