Add graceful shutdown in test_fault_tolerance

learning-at-home · Jun 9, 2024 · ffb639b · ffb639b
1 parent 784b7b9
commit ffb639b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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()