sanity

Lightning-AI · Jun 28, 2021 · 0d9284e · 0d9284e
1 parent 0b37832
commit 0d9284e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/plugins/test_ddp_plugin.py b/tests/plugins/test_ddp_plugin.py
@@ -75,4 +75,4 @@ def test_ddp_barrier_non_consecutive_device_ids(barrier_mock, tmpdir):
         accelerator="ddp",
     )
     trainer.fit(model)
-    barrier_mock.assert_called_with(device_ids=[gpus[trainer.local_rank]])
+    barrier_mock.assert_called_once_with(device_ids=[gpus[trainer.local_rank]])