From 3ae31ab0d76bb4bb3f7774ad642d2e0adac06b34 Mon Sep 17 00:00:00 2001 From: Oskar Taubert Date: Fri, 19 Apr 2024 15:49:44 +0200 Subject: [PATCH] fixed surrogate test problem, where workers would not wait for data to download on the second test --- tests/test_surrogate.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/test_surrogate.py b/tests/test_surrogate.py index 5d35fd75..2029b1c2 100644 --- a/tests/test_surrogate.py +++ b/tests/test_surrogate.py @@ -342,6 +342,8 @@ def test_mnist_static(mpi_tmp_path): logging_interval=1, # Logging interval debug=2, # Verbosity level ) + MPI.COMM_WORLD.barrier() + delattr(get_data_loaders, "barrier_called") @pytest.mark.mpi(min_size=4)