Fix trialkeeper flush (#2174)

4a07f9ed · SparkSnail · GitHub · 3c0ef842 · 4a07f9ed
Unverified Commit 4a07f9ed authored Mar 18, 2020 by SparkSnail Committed by GitHub Mar 18, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 7 additions and 0 deletions

tools/nni_trial_tool/log_utils.py tools/nni_trial_tool/log_utils.py +7 -0

No files found.
--- a/tools/nni_trial_tool/log_utils.py
+++ b/tools/nni_trial_tool/log_utils.py
@@ -85,6 +85,13 @@ class RemoteLogger(object):
        '''
        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