From 0091f54a91763284d1226e3577a3053738768f8a Mon Sep 17 00:00:00 2001
From: Antonio Bellotta <antonio.bellotta@epfl.ch>
Date: Wed, 13 Mar 2024 14:27:24 +0100
Subject: [PATCH] Hotfix to solve crashes when running on certain amount of
 ranks

---
 neurodamus/utils/memory.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/neurodamus/utils/memory.py b/neurodamus/utils/memory.py
index b0516cd9..0f09b099 100644
--- a/neurodamus/utils/memory.py
+++ b/neurodamus/utils/memory.py
@@ -408,7 +408,6 @@ def display_node_suggestions(self):
         logging.info("Please remember that it is suggested to use the same class of nodes "
                      "for both the dryrun and the actual simulation.")
 
-    @run_only_rank0
     def get_num_target_ranks(self, num_ranks):
         """
         Return the number of ranks to target for dry-run load balancing