fix channel slice

catalystneuro · Apr 30, 2024 · bb637c6 · bb637c6
1 parent 9c415fc
commit bb637c6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/common/models.py b/common/models.py
@@ -20,7 +20,7 @@ class RecordingContext(BaseModel):
     write_recording: bool = Field(default=False, description='Write recording')
     stub_test: bool = Field(default=False, description='Stub test')
     stub_test_duration_sec: float = Field(default=300, description='Stub test duration in seconds')
-    stub_test_channels: str = Field(default="1, 32", description='Stub test comma-separated range of channels')
+    stub_test_channels: str = Field(default="0, 31", description='Stub test comma-separated range of channels')
 
 
 class JobKwargs(BaseModel):

diff --git a/common/processor_pipeline.py b/common/processor_pipeline.py
@@ -51,7 +51,8 @@ def run_pipeline(context: PipelineFullContext):
         recording = recording.frame_slice(start_frame=0, end_frame=n_frames)
         # Limit the number of channels to the stub test range
         channel_ids = recording.channel_ids
-        stub_test_channels_ids = [channel_ids[int(a)] for a in context.recording_context.stub_test_channels.split(',')]
+        start, end = context.recording_context.stub_test_channels.split(',')
+        stub_test_channels_ids = [channel_ids[a] for a in range(int(start), int(end))]
         recording = recording.channel_slice(channel_ids=stub_test_channels_ids)
 
     logger.info(recording)