From ff7b1783fdb70153541d803da923d3d76aa45c4a Mon Sep 17 00:00:00 2001 From: dekken Date: Sat, 2 Aug 2025 17:21:17 +0200 Subject: [PATCH] move MPI size call after simulator runs --- tests/functional/harris/harris_2d.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/functional/harris/harris_2d.py b/tests/functional/harris/harris_2d.py index 86773443e..ba5249a9d 100644 --- a/tests/functional/harris/harris_2d.py +++ b/tests/functional/harris/harris_2d.py @@ -203,8 +203,6 @@ class HarrisTest(SimulatorTest): def __init__(self, *args, **kwargs): super(HarrisTest, self).__init__(*args, **kwargs) self.simulator = None - self.plot_dir = Path(f"{diag_dir}_plots") / str(cpp.mpi_size()) - self.plot_dir.mkdir(parents=True, exist_ok=True) def tearDown(self): super(HarrisTest, self).tearDown() @@ -217,7 +215,9 @@ def test_run(self): self.register_diag_dir_for_cleanup(diag_dir) Simulator(config()).run().reset() if cpp.mpi_rank() == 0: - plot(diag_dir, self.plot_dir) + plot_dir = Path(f"{diag_dir}_plots") / str(cpp.mpi_size()) + plot_dir.mkdir(parents=True, exist_ok=True) + plot(diag_dir, plot_dir) cpp.mpi_barrier() return self