From e6ed09664fef7daf39217147aa47fbab9f044032 Mon Sep 17 00:00:00 2001
From: allegroai <>
Date: Wed, 25 Sep 2019 00:32:08 +0300
Subject: [PATCH] Fix force flush logs at exit

---
 trains/backend_interface/metrics/reporter.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/trains/backend_interface/metrics/reporter.py b/trains/backend_interface/metrics/reporter.py
index 9168539e..c549036e 100644
--- a/trains/backend_interface/metrics/reporter.py
+++ b/trains/backend_interface/metrics/reporter.py
@@ -86,6 +86,10 @@ class Reporter(InterfaceBase, AbstractContextManager, SetupUploadMixin, AsyncMan
             # wait for all reports
             if self.get_num_results() > 0:
                 self.wait_for_results()
+        # make sure we flushed everything
+        self._write()
+        if self.get_num_results() > 0:
+            self.wait_for_results()
 
     def _report(self, ev):
         self._events.append(ev)