Increase timeouts and number of processes

learning-at-home · Oct 19, 2024 · f104cd6 · f104cd6
1 parent d789ff6
commit f104cd6
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/.github/workflows/run-tests.yml b/.github/workflows/run-tests.yml
@@ -39,7 +39,7 @@ jobs:
         run: |
           cd tests
           export HIVEMIND_MEMORY_SHARING_STRATEGY=file_descriptor
-          pytest --durations=0 --durations-min=1.0 -n auto -v
+          pytest --durations=0 --durations-min=1.0 -n 4 -v
   build_and_test_p2pd:
     runs-on: ubuntu-latest
     timeout-minutes: 10

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=1.0,
-            min_matchmaking_time=1.0,
-            next_chunk_timeout=0.5,
-            allreduce_timeout=10,
+            request_timeout=1.5,
+            min_matchmaking_time=2.0,
+            next_chunk_timeout=1.0,
+            allreduce_timeout=15,
             part_size_bytes=2**16,
             client_mode=(i == 1),
             start=True,