Increase timeouts in test_fault_tolerance

learning-at-home · Oct 19, 2024 · d789ff6 · d789ff6
1 parent 647224d
commit d789ff6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_allreduce_fault_tolerance.py b/tests/test_allreduce_fault_tolerance.py
@@ -153,10 +153,10 @@ def _make_tensors():
             _make_tensors(),
             hivemind.DHT(initial_peers=dht.get_visible_maddrs(), start=True),
             prefix="test",
-            request_timeout=0.3,
+            request_timeout=1.0,
             min_matchmaking_time=1.0,
             next_chunk_timeout=0.5,
-            allreduce_timeout=5,
+            allreduce_timeout=10,
             part_size_bytes=2**16,
             client_mode=(i == 1),
             start=True,