fix black

facebookresearch · Apr 29, 2024 · 2677170 · 2677170
1 parent 364d73f
commit 2677170
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/train/compute/python/tools/execution_trace.py b/train/compute/python/tools/execution_trace.py
@@ -104,6 +104,7 @@ def is_leaf_tensor(self):
 @dataclass
 class _CommArgs:
     """Contains communication collective metadata"""
+
     collective_name: str
     dtype: str
     # .. TODO add more see https://github.com/pytorch/pytorch/issues/124674

diff --git a/train/compute/python/tools/validate_trace.py b/train/compute/python/tools/validate_trace.py
@@ -13,7 +13,6 @@
 
 
 class TraceValidator:
-
     def __init__(self, execution_trace: ExecutionTrace):
         self.et = execution_trace
 
@@ -60,7 +59,7 @@ def check_comms_node_old(n) -> bool:
 
         def check_comms_node_new(n) -> bool:
             """New elements are added as per
-                https://github.com/pytorch/pytorch/issues/124674
+            https://github.com/pytorch/pytorch/issues/124674
             """
             # TODO check for node.commArgs dataclass
             print(n.commArgs)