Revert changes to test_progress_tracker

learning-at-home · Jul 14, 2024 · a0fbbc9 · a0fbbc9
1 parent d321a79
commit a0fbbc9
Showing 1 changed file with 5 additions and 7 deletions.
diff --git a/tests/test_optimizer.py b/tests/test_optimizer.py
@@ -241,15 +241,13 @@ def run_worker(index: int, batch_size: int, step_time: float, initial_peers: Lis
             min_refresh_period=0.1,
             default_refresh_period=0.2,
             max_refresh_period=0.5,
-            private_key=RSAPrivateKey(),
         )
-        with tracker.pause_updates():
-            barrier.wait()
-            if index == 4:
-                delayed_start_evt.wait()
+        barrier.wait()
+        if index == 4:
+            delayed_start_evt.wait()
 
-            local_epoch = 2 if index == 4 else 0
-            samples_accumulated = 0
+        local_epoch = 2 if index == 4 else 0
+        samples_accumulated = 0
 
         while True:
             time.sleep(step_time)