From ffb639b978126a7618185bddba517cd9e955cad9 Mon Sep 17 00:00:00 2001 From: Max Ryabinin Date: Sun, 9 Jun 2024 22:07:50 +0100 Subject: [PATCH] Add graceful shutdown in test_fault_tolerance --- tests/test_allreduce_fault_tolerance.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/test_allreduce_fault_tolerance.py b/tests/test_allreduce_fault_tolerance.py index d1ee66d99..12e310eba 100644 --- a/tests/test_allreduce_fault_tolerance.py +++ b/tests/test_allreduce_fault_tolerance.py @@ -209,3 +209,4 @@ def _make_tensors(): for averager in averagers: averager.shutdown() + dht.shutdown()