From 745f33efea14dcfffc17fcdea8dc2188ee9563f7 Mon Sep 17 00:00:00 2001 From: Chien-Chin Huang Date: Fri, 10 Jan 2025 11:11:07 -0800 Subject: [PATCH] Update [ghstack-poisoned] --- torchft/process_group_test.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/torchft/process_group_test.py b/torchft/process_group_test.py index 95e12d7..f6a89d6 100644 --- 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")