Feat: support custom node sets defined in simulation config (#192)

BlueBrain · Jul 9, 2024 · 2e672f7 · 2e672f7
1 parent e1a7f68
commit 2e672f7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/bluecellulab/circuit/circuit_access/sonata_circuit_access.py b/bluecellulab/circuit/circuit_access/sonata_circuit_access.py
@@ -47,7 +47,7 @@ def __init__(self, simulation_config: str | Path | SimulationConfig) -> None:
             raise FileNotFoundError(f"Circuit config file {simulation_config} not found.")
 
         if isinstance(simulation_config, SonataSimulationConfig):
-            self.config: SimulationConfig = simulation_config
+            self.config = simulation_config
         else:
             self.config = SonataSimulationConfig(simulation_config)
         circuit_config = self.config.impl.config["network"]
@@ -191,7 +191,7 @@ def is_valid_group(self, group: str) -> bool:
 
     @lru_cache(maxsize=16)
     def get_target_cell_ids(self, target: str) -> set[CellId]:
-        ids = self._circuit.nodes.ids(target)
+        ids = self._circuit.nodes.ids(self.config.impl.node_sets[target])
         return {CellId(x.population, x.id) for x in ids}
 
     @lru_cache(maxsize=100)