From 740727462cb6cd0d078e79e5548ddd982d0cb600 Mon Sep 17 00:00:00 2001
From: Jorge Blanco Alonso <jorge.blancoalonso@epfl.ch>
Date: Wed, 10 Apr 2024 14:34:18 +0200
Subject: [PATCH] Add offset to getSpGid in case of load balance

---
 neurodamus/cell_distributor.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/neurodamus/cell_distributor.py b/neurodamus/cell_distributor.py
index 38e72492..65a93fb7 100644
--- a/neurodamus/cell_distributor.py
+++ b/neurodamus/cell_distributor.py
@@ -506,7 +506,8 @@ def getSpGid(self, gid):
         """
         manager = self._find_manager(gid)
         if manager._binfo:
-            return manager._binfo.thishost_gid(gid - manager.local_nodes.offset)
+            gid_offset = manager.local_nodes.offset
+            return manager._binfo.thishost_gid(gid - gid_offset) + gid_offset
         return gid
 
     def getPopulationInfo(self, gid):