diff --git a/common/models.py b/common/models.py index f05c632..2808149 100644 --- 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 index 197e4b7..f1d16f4 100644 --- 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)