From 7e5d6e2e93774821f14048f6df8a9cfcfc57eb9c Mon Sep 17 00:00:00 2001 From: Henry Tsang Date: Mon, 2 Oct 2023 11:44:02 -0700 Subject: [PATCH] Back out "Merge logs torchrec/distributed/planner/stats.py" Summary: Original commit changeset: 347dfb3e5fd7 Original Phabricator Diff: D48619298 Differential Revision: D49835790 --- 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,