Skip to content

Commit

Permalink
Revert of D48619298 (pytorch#1419)
Browse files Browse the repository at this point in the history
Summary:
Pull Request resolved: pytorch#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
  • Loading branch information
Aayush Gupta authored and facebook-github-bot committed Oct 3, 2023
1 parent 420a691 commit 297fea8
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions torchrec/distributed/planner/stats.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down Expand Up @@ -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,
Expand Down

0 comments on commit 297fea8

Please sign in to comment.