Update on_finish from async to sync (tensorflow#5242)

yhliang2018 · web-flow · commit e0f6a3928ec3 · 2018-09-04T16:52:46.000-07:00
diff --git a/official/utils/logs/logger.py b/official/utils/logs/logger.py
@@ -291,12 +291,11 @@ def log_run_info(self, model_name, dataset_name, run_params, test_id=None):
          RUN_STATUS_RUNNING))
 
   def on_finish(self, status):
-    thread.start_new_thread(
-        self._bigquery_uploader.update_run_status,
-        (self._bigquery_data_set,
-         self._bigquery_run_status_table,
-         self._run_id,
-         status))
+    self._bigquery_uploader.update_run_status(
+        self._bigquery_data_set,
+        self._bigquery_run_status_table,
+        self._run_id,
+        status)
 
 
 def _gather_run_info(model_name, dataset_name, run_params, test_id):
