Browse Source

Fix spiketrains ending up in in block.groups

Julia Sprenger 3 years ago
parent
commit
12e4548b3c
1 changed files with 3 additions and 3 deletions
  1. 3 3
      code/reachgraspio/reachgraspio.py

+ 3 - 3
code/reachgraspio/reachgraspio.py

@@ -966,11 +966,11 @@ class ReachGraspIO(BlackrockIO):
                                                           coordinates_y=coordinates[1]))
 
             # updating annotations of groups and spiketrains
-            objs = block.groups
+            sts = []
             for seg in block.segments:
-                objs.extend(seg.spiketrains)
+                sts.extend(seg.spiketrains)
 
-            for obj in objs:
+            for obj in sts + block.groups:
                 if 'channel_id' in obj.annotations:
                     chid = obj.annotations['channel_id']
                     ca_id, *coordinates = self.__convert_chids_and_coordinates([chid])