Skip to content

Commit 4567c1f

Browse files
authored
channel loader returns the electrode sites indexing, closes #893 (#894)
1 parent 1382460 commit 4567c1f

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

brainbox/io/one.py

+4-1
Original file line numberDiff line numberDiff line change
@@ -1017,7 +1017,10 @@ def load_channels(self, **kwargs):
10171017
self.download_spike_sorting_object(obj='channels', missing='ignore', **kwargs)
10181018
channels = self._load_object(self.files['channels'], wildcards=self.one.wildcards)
10191019
if 'electrodeSites' in self.files: # if common dict keys, electrodeSites prevails
1020-
channels = channels | self._load_object(self.files['electrodeSites'], wildcards=self.one.wildcards)
1020+
esites = channels | self._load_object(self.files['electrodeSites'], wildcards=self.one.wildcards)
1021+
if alfio.check_dimensions(esites) != 0:
1022+
esites = self._load_object(self.files['electrodeSites'], wildcards=self.one.wildcards)
1023+
esites['rawInd'] = np.arange(esites[list(esites.keys())[0]].shape[0])
10211024
if 'brainLocationIds_ccf_2017' not in channels:
10221025
_logger.debug(f"loading channels from alyx for {self.files['channels']}")
10231026
_channels, self.histology = _load_channel_locations_traj(

0 commit comments

Comments
 (0)