blacken

Author: Jackmin801

torchft/lighthouse_test.py

@@ -4,6 +4,7 @@
 from torchft import Manager, ProcessGroupGloo
 import time
 
+
 class TestLighthouse(TestCase):
     def test_join_timeout_behavior(self) -> None:
         """Test that join_timeout_ms affects joining behavior"""
@@ -14,7 +15,7 @@ def test_join_timeout_behavior(self) -> None:
             min_replicas=1,
             join_timeout_ms=100,
         )
-        
+
         # Create a manager that tries to join
         try:
             store = dist.TCPStore(
@@ -37,7 +38,7 @@ def test_join_timeout_behavior(self) -> None:
                 use_async_quorum=False,
                 lighthouse_addr=lighthouse.address(),
             )
-            
+
             start_time = time.time()
             manager.start_quorum()
             time_taken = time.time() - start_time
@@ -46,15 +47,15 @@ def test_join_timeout_behavior(self) -> None:
         finally:
             # Cleanup
             lighthouse.shutdown()
-            if 'manager' in locals():
+            if "manager" in locals():
                 manager.shutdown()
-    
+
         lighthouse = Lighthouse(
             bind="[::]:0",
             min_replicas=1,
             join_timeout_ms=400,
         )
-        
+
         # Create a manager that tries to join
         try:
             store = dist.TCPStore(
@@ -77,7 +78,7 @@ def test_join_timeout_behavior(self) -> None:
                 use_async_quorum=False,
                 lighthouse_addr=lighthouse.address(),
             )
-            
+
             start_time = time.time()
             manager.start_quorum()
             time_taken = time.time() - start_time
@@ -86,6 +87,5 @@ def test_join_timeout_behavior(self) -> None:
         finally:
             # Cleanup
             lighthouse.shutdown()
-            if 'manager' in locals():
+            if "manager" in locals():
                 manager.shutdown()
-