Skip to content

GitLab

Unverified Commit e0f6a392 authored by Yanhui Liang's avatar Yanhui Liang Committed by GitHub
Browse files

Update on_finish from async to sync (#5242)

parent 0783f1cf
Show whitespace changes
Inline Side-by-side
Showing with 5 additions and 6 deletions
official/utils/logs/logger.py
View file @ e0f6a392
...@@ -291,12 +291,11 @@ class BenchmarkBigQueryLogger(BaseBenchmarkLogger): ...@@ -291,12 +291,11 @@ class BenchmarkBigQueryLogger(BaseBenchmarkLogger):
RUN_STATUS_RUNNING)) RUN_STATUS_RUNNING))
def on_finish(self, status): def on_finish(self, status):
thread.start_new_thread( self._bigquery_uploader.update_run_status(
self._bigquery_uploader.update_run_status, self._bigquery_data_set,
(self._bigquery_data_set,
self._bigquery_run_status_table, self._bigquery_run_status_table,
self._run_id, self._run_id,
status)) status)
def _gather_run_info(model_name, dataset_name, run_params, test_id): def _gather_run_info(model_name, dataset_name, run_params, test_id):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment