Skip to content

Commit 389f54d

Browse files
authored
Merge pull request #15 from cwindolf/iblsorter
peeler empty result should have the same structure as a normal chunk
2 parents df65ce9 + 04a2371 commit 389f54d

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/dartsort/transform/temporal_pca.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -94,7 +94,7 @@ def needs_fit(self):
9494
return self._needs_fit
9595

9696
def _temporal_slice(self, waveforms):
97-
if self.temporal_slice is None:
97+
if getattr(self, 'temporal_slice', None) is None:
9898
return waveforms
9999

100100
return waveforms[:, self.temporal_slice]
@@ -177,6 +177,7 @@ class TemporalPCAFeaturizer(BaseWaveformFeaturizer, BaseTemporalPCA):
177177

178178
def transform(self, waveforms, max_channels, channel_index=None, return_in_probe=False):
179179
waveforms = self._temporal_slice(waveforms)
180+
180181
if channel_index is None:
181182
channel_index = self.channel_index
182183
(

0 commit comments

Comments
 (0)