Propagate gap_tolerance_ms from Blackrock neo reader (#4453)

Author: h-mayorquin

src/spikeinterface/extractors/neoextractors/blackrock.py

@@ -24,6 +24,11 @@ class BlackrockRecordingExtractor(NeoBaseRecordingExtractor):
     use_names_as_ids : bool, default: False
         Determines the format of the channel IDs used by the extractor. If set to True, the channel IDs will be the
         names from NeoRawIO. If set to False, the channel IDs will be the ids provided by NeoRawIO.
+    gap_tolerance_ms : float | None, default: None
+        Maximum acceptable gap size in milliseconds for automatic segmentation.
+        If None (default), an error is raised when timestamp gaps are detected.
+        If a value is provided, gaps smaller than this threshold are ignored and
+        gaps larger than this threshold create new segments.
 
     """
 
@@ -36,8 +41,11 @@ def __init__(
         stream_name=None,
         all_annotations: bool = False,
         use_names_as_ids: bool = False,
+        gap_tolerance_ms: float | None = None,
     ):
         neo_kwargs = self.map_to_neo_kwargs(file_path)
+        if gap_tolerance_ms is not None:
+            neo_kwargs["gap_tolerance_ms"] = gap_tolerance_ms
         neo_kwargs["load_nev"] = False  # Avoid loading spikes release in neo 0.12.0
 
         # trick to avoid to select automatically the correct stream_id
@@ -52,7 +60,7 @@ def __init__(
             use_names_as_ids=use_names_as_ids,
             **neo_kwargs,
         )
-        self._kwargs.update({"file_path": str(Path(file_path).absolute())})
+        self._kwargs.update({"file_path": str(Path(file_path).absolute()), "gap_tolerance_ms": gap_tolerance_ms})
 
     @classmethod
     def map_to_neo_kwargs(cls, file_path):
@@ -82,6 +90,11 @@ class BlackrockSortingExtractor(NeoBaseSortingExtractor):
     nsx_to_load : int | list | str, default: None
         IDs of nsX file from which to load data, e.g., if set to 5 only data from the ns5 file are loaded.
         If 'all', then all nsX will be loaded. If None, all nsX files will be loaded. If empty list, no nsX files will be loaded.
+    gap_tolerance_ms : float | None, default: None
+        Maximum acceptable gap size in milliseconds for automatic segmentation.
+        If None (default), an error is raised when timestamp gaps are detected.
+        If a value is provided, gaps smaller than this threshold are ignored and
+        gaps larger than this threshold create new segments.
     """
 
     NeoRawIOClass = "BlackrockRawIO"
@@ -94,8 +107,11 @@ def __init__(
         stream_name: str | None = None,
         sampling_frequency: float | None = None,
         nsx_to_load: int | list | str | None = None,
+        gap_tolerance_ms: float | None = None,
     ):
         neo_kwargs = self.map_to_neo_kwargs(file_path)
+        if gap_tolerance_ms is not None:
+            neo_kwargs["gap_tolerance_ms"] = gap_tolerance_ms
         NeoBaseSortingExtractor.__init__(
             self,
             stream_id=stream_id,
@@ -110,6 +126,7 @@ def __init__(
             "sampling_frequency": sampling_frequency,
             "stream_id": stream_id,
             "stream_name": stream_name,
+            "gap_tolerance_ms": gap_tolerance_ms,
         }
 
     @classmethod