Strip lines when log-tailing

Differential Revision: D66191123 Pull Request resolved: #973
pytorch · Dec 3, 2024 · cb0e457 · cb0e457
1 parent a5f840e
commit cb0e457
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchx/util/log_tee_helpers.py b/torchx/util/log_tee_helpers.py
@@ -102,7 +102,7 @@ def _print_log_lines_for_role_replica(
                 color_begin = ""
                 color_end = ""
             prefix = f"{color_begin}{which_role}/{which_replica}{color_end} "
-            print(_prefix_line(prefix, line), file=dst, end="", flush=True)
+            print(_prefix_line(prefix, line.strip()), file=dst, end="\n", flush=True)
     except Exception as e:
         exceptions.put(e)
         raise