File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 8383 get_global_job_kwargs ,
8484 set_global_job_kwargs ,
8585 reset_global_job_kwargs ,
86+ is_set_global_job_kwargs_set ,
8687)
8788
8889# tools
Original file line number Diff line number Diff line change 33import numpy as np
44
55from spikeinterface .core .core_tools import define_function_handling_dict_from_class
6- from spikeinterface .core import get_chunk_with_margin , ensure_chunk_size , get_global_job_kwargs
6+ from spikeinterface .core import (
7+ get_chunk_with_margin ,
8+ ensure_chunk_size ,
9+ get_global_job_kwargs ,
10+ is_set_global_job_kwargs_set ,
11+ )
712
813from .basepreprocessor import BasePreprocessor , BasePreprocessorSegment
914
@@ -118,7 +123,7 @@ def __init__(
118123 margin = int (margin_ms * fs / 1000.0 )
119124
120125 global_job_kwargs_chunk_size = ensure_chunk_size (recording , ** get_global_job_kwargs ())
121- if margin > MARGIN_TO_CHUNK_PERCENT_WARNING * global_job_kwargs_chunk_size :
126+ if is_set_global_job_kwargs_set () and margin > MARGIN_TO_CHUNK_PERCENT_WARNING * global_job_kwargs_chunk_size :
122127 warnings .warn (
123128 f"The margin size ({ margin } samples) is more than { int (MARGIN_TO_CHUNK_PERCENT_WARNING * 100 )} % "
124129 f"of the global chunk size { global_job_kwargs_chunk_size } samples. This may lead to performance bottlenecks when "
You can’t perform that action at this time.
0 commit comments