Update

[ghstack-poisoned]
pytorch · Jan 10, 2025 · 745f33e · 745f33e
1 parent 31ba2ef
commit 745f33e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/torchft/process_group_test.py b/torchft/process_group_test.py
@@ -361,7 +361,9 @@ def _test_init_device_mesh(world_size: int, rank: int) -> None:
             not isinstance(device_mesh.get_group("dp_shard"), ManagedProcessGroup)
         )
         replicate_group = device_mesh.get_group("dp_replicate")
-        testcase.assertEqual(cast(ManagedProcessGroup, replicate_group)._manager, manager)
+        testcase.assertEqual(
+            cast(ManagedProcessGroup, replicate_group)._manager, manager
+        )
         replicate_mesh = device_mesh["dp_replicate"]
         testcase.assertEqual(replicate_mesh.get_group(), replicate_group)
         flatten_mesh = device_mesh._flatten("dp")