Fixes lint

syed-ahmed · syed-ahmed · commit 1f553e01dcf4 · 2023-04-18T09:47:29.000-07:00
diff --git a/torch/nn/parallel/distributed.py b/torch/nn/parallel/distributed.py
@@ -749,12 +749,11 @@ def __init__(
         else:
             self.process_group = process_group
 
-        if (
-            dist.get_backend(self.process_group) == "nccl"
-            and torch.cuda.nccl.version() >= (2, 17)
-        ):
+        if dist.get_backend(
+            self.process_group
+        ) == "nccl" and torch.cuda.nccl.version() >= (2, 17):
             # Note: NVIDIA recommends using CGA Cluster Size of 2 when using DDP.
-            default_cga = dist.ProcessGroupNCCL.Options().config.cga_cluster_size # type: ignore[attr-defined]
+            default_cga = dist.ProcessGroupNCCL.Options().config.cga_cluster_size  # type: ignore[attr-defined]
             default_pg_nccl = self.process_group._get_backend(torch.device("cuda"))
             current_cga = default_pg_nccl.options.config.cga_cluster_size
             if current_cga == default_cga:
