use sync context for the complete reporting

Author: geyang

ml_logger/VERSION

@@ -1 +1 @@
-0.8.31
+0.8.32

ml_logger/ml_logger/__init__.py

@@ -186,10 +186,12 @@ def thunk(*args, **kwargs):
 
             results = fn(*(args or ARGS), **_KWARGS)
 
-            logger.log_line("========== execution is complete ==========")
-            logger.job_completed()
-            logger.flush()
+            with logger.SyncContext():  # Make sure uploaded finished before termination.
+                logger.log_line("========== execution is complete ==========")
+                logger.job_completed()
+                logger.flush()
             time.sleep(3)
+
         except Exception as e:
             tb = traceback.format_exc()
             with logger.SyncContext():  # Make sure uploaded finished before termination.