Skip to content

Commit e977f36

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 8f167fc commit e977f36

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/spikeinterface/core/waveform_tools.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -592,6 +592,7 @@ def _init_worker_distribute_single_buffer(
592592
worker_dict["all_waveforms"] = all_waveforms
593593
elif mode == "shared_memory":
594594
from multiprocessing.shared_memory import SharedMemory
595+
595596
shm_name, dtype, shape = wf_array_info["shm_name"], wf_array_info["dtype"], wf_array_info["shape"]
596597
shm = SharedMemory(shm_name)
597598
all_waveforms = np.ndarray(shape=shape, dtype=dtype, buffer=shm.buf)
@@ -660,6 +661,7 @@ def _worker_distribute_single_buffer(segment_index, start_frame, end_frame, work
660661
if worker_dict["mode"] == "memmap":
661662
all_waveforms.flush()
662663

664+
663665
def split_waveforms_by_units(unit_ids, spikes, all_waveforms, sparsity_mask=None, folder=None):
664666
"""
665667
Split a single buffer waveforms into waveforms by units (multi buffers or multi files).

0 commit comments

Comments
 (0)