Fix

spyking-circus · Dec 13, 2019 · 1253e82 · 1253e82
1 parent 4ffa6fd
commit 1253e82
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/circus/clustering.py b/circus/clustering.py
@@ -98,7 +98,7 @@ def main(params, nb_cpu, nb_gpu, use_gpu):
         search_peaks = ['neg', 'pos']
 
     nodes_indices = {}
-    for elec in nodes:
+    for elec in numpy.arange(N_e):
         nodes_indices[elec] = inv_nodes[edges[nodes[elec]]]
 
     smart_searches = {}

diff --git a/circus/whitening.py b/circus/whitening.py
@@ -56,7 +56,7 @@ def main(params, nb_cpu, nb_gpu, use_gpu):
         print_and_log(["Analyzing data to get whitening matrices and thresholds..."], 'default', logger)
 
     nodes_indices = {}
-    for elec in nodes:
+    for elec in numpy.arange(N_e):
         nodes_indices[elec] = inv_nodes[edges[nodes[elec]]]
 
     if use_gpu: