Minor fix for node selection

BlueBrain · Jul 2, 2024 · a451732 · a451732
1 parent 758e740
commit a451732
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/CHANGELOG.rst b/CHANGELOG.rst
@@ -7,7 +7,7 @@ Version 0.0.11.dev1
 New Features
 ~~~~~~~~~~~~
 - New synapse position re-use mode "reuse_strict" with re-use restricted to source selection
-- Minor fix for empty data splits
+- Minor fixes for empty data splits and node selection
 
 
 Version 0.0.11.dev0

diff --git a/connectome_manipulator/access_functions.py b/connectome_manipulator/access_functions.py
@@ -210,6 +210,8 @@ def get_node_ids(nodes, sel_spec, split_ids=None):
             log.log_assert(isinstance(node_set, str), "Node set must be a string!")
             if selection is None:  # Nothing else selected
                 gids = nodes.ids(node_set)
+                if split_ids is not None:
+                    gids = np.intersect1d(gids, sel_ids.flatten().astype(np.int64))
             else:  # Otherwise, intersect with selection
                 gids = np.intersect1d(gids, nodes.ids(node_set))
     else: