@@ -333,10 +333,10 @@ class TransformSorting(BaseSorting):
333333 def __init__ (
334334 self ,
335335 sorting : BaseSorting ,
336- added_spikes_existing_units : np .array | None = None ,
337- added_spikes_new_units : np .array | None = None ,
338- new_unit_ids : list [str | int ] | None = None ,
339- refractory_period_ms : float | None = None ,
336+ added_spikes_existing_units : np .array | None = None ,
337+ added_spikes_new_units : np .array | None = None ,
338+ new_unit_ids : list [str | int ] | None = None ,
339+ refractory_period_ms : float | None = None ,
340340 ):
341341 sampling_frequency = sorting .get_sampling_frequency ()
342342 unit_ids = list (sorting .get_unit_ids ())
@@ -1089,11 +1089,11 @@ def __init__(
10891089 self ,
10901090 num_channels : int ,
10911091 sampling_frequency : float ,
1092- durations : list [float ],
1093- noise_levels : float | np .array = 1.0 ,
1094- cov_matrix : np .array | None = None ,
1095- dtype : np .dtype | str | None = "float32" ,
1096- seed : int | None = None ,
1092+ durations : list [float ],
1093+ noise_levels : float | np .array = 1.0 ,
1094+ cov_matrix : np .array | None = None ,
1095+ dtype : np .dtype | str | None = "float32" ,
1096+ seed : int | None = None ,
10971097 strategy : Literal ["tile_pregenerated" , "on_the_fly" ] = "tile_pregenerated" ,
10981098 noise_block_size : int = 30000 ,
10991099 ):
@@ -1708,11 +1708,11 @@ def __init__(
17081708 self ,
17091709 sorting : BaseSorting ,
17101710 templates : np .ndarray ,
1711- nbefore : list [int ] | int | None = None ,
1712- amplitude_factor : list [float ] | float | None = None ,
1713- parent_recording : BaseRecording | None = None ,
1714- num_samples : list [int ] | int | None = None ,
1715- upsample_vector : np .array | None = None ,
1711+ nbefore : list [int ] | int | None = None ,
1712+ amplitude_factor : list [float ] | float | None = None ,
1713+ parent_recording : BaseRecording | None = None ,
1714+ num_samples : list [int ] | int | None = None ,
1715+ upsample_vector : np .array | None = None ,
17161716 check_borders : bool = False ,
17171717 ) -> None :
17181718 templates = np .asarray (templates )
@@ -1844,10 +1844,10 @@ def __init__(
18441844 spike_vector : np .ndarray ,
18451845 templates : np .ndarray ,
18461846 nbefore : int ,
1847- amplitude_vector : list [float ] | None ,
1848- upsample_vector : list [float ] | None ,
1849- parent_recording_segment : BaseRecordingSegment | None = None ,
1850- num_samples : int | None = None ,
1847+ amplitude_vector : list [float ] | None ,
1848+ upsample_vector : list [float ] | None ,
1849+ parent_recording_segment : BaseRecordingSegment | None = None ,
1850+ num_samples : int | None = None ,
18511851 ) -> None :
18521852 BaseRecordingSegment .__init__ (
18531853 self ,
@@ -1867,9 +1867,9 @@ def __init__(
18671867
18681868 def get_traces (
18691869 self ,
1870- start_frame : int | None = None ,
1871- end_frame : int | None = None ,
1872- channel_indices : list | None = None ,
1870+ start_frame : int | None = None ,
1871+ end_frame : int | None = None ,
1872+ channel_indices : list | None = None ,
18731873 ) -> np .ndarray :
18741874 if channel_indices is None :
18751875 n_channels = self .templates .shape [2 ]
0 commit comments