Merge pull request #76 from microsoft/master

Fix trialkeeper flush (microsoft#2174)
chicm-ms · Mar 18, 2020 · 8b5613a · 8b5613a
2 parents 0439bc1 + 4a07f9e
commit 8b5613a
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/tools/nni_trial_tool/log_utils.py b/tools/nni_trial_tool/log_utils.py
@@ -85,6 +85,13 @@ def get_pipelog_reader(self):
         '''
         return PipeLogReader(self.logger, self.log_collection, logging.INFO)
 
+    def flush(self):
+        '''
+        Add flush in handler
+        '''
+        for handler in self.logger.handlers:
+            handler.flush()
+
     def write(self, buf):
         '''
         Write buffer data into logger/stdout