From a0bc65b338dab5d0c2363cdb36f6964b9b390e7f Mon Sep 17 00:00:00 2001 From: Aayush Gupta Date: Mon, 2 Oct 2023 16:38:51 -0700 Subject: [PATCH] Revert of D48619298 Summary: 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 Differential Revision: D49850102 --- 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,