Update on_finish from async to sync (#5242)

e0f6a392 · Yanhui Liang · GitHub · 0783f1cf · e0f6a392
Unverified Commit e0f6a392 authored Sep 04, 2018 by Yanhui Liang Committed by GitHub Sep 04, 2018
Show whitespace changes
Inline Side-by-side

Showing with 5 additions and 6 deletions

official/utils/logs/logger.py official/utils/logs/logger.py +5 -6

No files found.
--- a/official/utils/logs/logger.py
+++ b/official/utils/logs/logger.py
@@ -291,12 +291,11 @@ class BenchmarkBigQueryLogger(BaseBenchmarkLogger):
         RUN_STATUS_RUNNING))

  def on_finish(self, status):
-    thread.start_new_thread(
-        self._bigquery_uploader.update_run_status,
-        (self._bigquery_data_set,
+    self._bigquery_uploader.update_run_status(
+        self._bigquery_data_set,
        self._bigquery_run_status_table,
        self._run_id,
-         status))
+        status)


 def _gather_run_info(model_name, dataset_name, run_params, test_id):