From 0eb0abc25bdc295ddfce67c1d49c53706ef7104e Mon Sep 17 00:00:00 2001
From: Kael Dai <kael.dai@gmail.com>
Date: Mon, 11 May 2026 09:11:15 -0700
Subject: [PATCH] fixing issue with node_ids as a list instead of a dict

---
 bmtk/simulator/core/node_sets.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/bmtk/simulator/core/node_sets.py b/bmtk/simulator/core/node_sets.py
index a2ef6e1a..8eea1081 100644
--- a/bmtk/simulator/core/node_sets.py
+++ b/bmtk/simulator/core/node_sets.py
@@ -8,8 +8,9 @@ def __init__(self, filter_params, network):
         self._preselected_gids = None
 
         if isinstance(filter_params, list):
-            self._preselected_gids = filter_params
-        elif isinstance(filter_params, dict):
+            filter_params = {'node_id': filter_params}
+       
+        if isinstance(filter_params, dict):
             self._filter = filter_params.copy()
             self._populations = self._find_populations()
         else:
@@ -34,6 +35,9 @@ def population_names(self):
 
     def gids(self):
         if self._preselected_gids is not None:
+            # print('HERE')
+            # print(self.population_names())
+            # exit()
             for gid in self._preselected_gids:
                 yield gid
         else: