mute verbose

Lightning-AI · Mar 31, 2020 · a17f217 · a17f217
1 parent 987df24
commit a17f217
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/pytorch_lightning/profiler/profilers.py b/pytorch_lightning/profiler/profilers.py
@@ -115,9 +115,7 @@ def __init__(self, output_filename: str = None):
         self.output_fname = output_filename
         self.output_file = open(self.output_fname, 'w') if self.output_fname else None
 
-        streaming_out = [log.info]
-        if self.output_file:
-            streaming_out.append(self.output_file.write)
+        streaming_out = [self.output_file.write] if self.output_file else [log.info]
         super().__init__(output_streams=streaming_out)
 
     def start(self, action_name: str) -> None:
@@ -186,9 +184,7 @@ def __init__(self, output_filename: str = None, line_count_restriction: float =
         self.output_fname = output_filename
         self.output_file = open(self.output_fname, 'w') if self.output_fname else None
 
-        streaming_out = [log.info]
-        if self.output_file:
-            streaming_out.append(self.output_file.write)
+        streaming_out = [self.output_file.write] if self.output_file else [log.info]
         super().__init__(output_streams=streaming_out)
 
     def start(self, action_name: str) -> None: