Fix trialkeeper flush (#2174) (#2177)

Co-authored-by: SparkSnail <shinaiyang@pku.edu.cn>
microsoft · Mar 18, 2020 · faf3e19 · faf3e19
1 parent c358adc
commit faf3e19
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