Set join=False for mp_spawn in the ddp_test

Lightning-AI · Aug 29, 2021 · f23451e · f23451e
1 parent 64e0863
commit f23451e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/text/test_bertscore.py b/tests/text/test_bertscore.py
@@ -315,4 +315,4 @@ def _test_score_ddp_fn(rank, world_size, preds, refs):
 def test_score_ddp(preds, refs):
     """Tests for metric using DDP."""
     world_size = 2
-    mp.spawn(_test_score_ddp_fn, args=(world_size, preds, refs), nprocs=world_size, join=True)
+    mp.spawn(_test_score_ddp_fn, args=(world_size, preds, refs), nprocs=world_size, join=False)