From 297fea8ca18bfe081123ee5c1c8ad5926d4abdf3 Mon Sep 17 00:00:00 2001 From: Aayush Gupta Date: Mon, 2 Oct 2023 20:05:53 -0700 Subject: [PATCH] Revert of D48619298 (#1419) Summary: Pull Request resolved: https://github.com/pytorch/torchrec/pull/1419 Folks are complaining about how this new format is much harder to get to the lines we care about. None of the existing solutions are at par with the original experience. Details here: https://fb.workplace.com/groups/1302733780185462/posts/1765846840540818 Reviewed By: henrylhtsang Differential Revision: D49850102 fbshipit-source-id: 8e789865f799a953cbd08dfbcde1ad0943e1901e --- torchrec/distributed/planner/stats.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/torchrec/distributed/planner/stats.py b/torchrec/distributed/planner/stats.py index d1106a5b3..ef869b967 100644 --- a/torchrec/distributed/planner/stats.py +++ b/torchrec/distributed/planner/stats.py @@ -6,7 +6,6 @@ # LICENSE file in the root directory of this source tree. import logging -import os from collections import defaultdict from typing import Any, cast, Dict, List, Optional, Tuple, Union @@ -391,7 +390,8 @@ def log( self._stats_table.append("#" * self._width) - logger.info(os.linesep + os.linesep.join(self._stats_table)) + for row in self._stats_table: + logger.info(row) def _get_shard_stats( self,