fixed surrogate test problem, where workers would not wait for data t…

…o download on the second test
Helmholtz-AI-Energy · Apr 19, 2024 · 3ae31ab · 3ae31ab
1 parent a0a90ee
commit 3ae31ab
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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)