From 1d2401e6e94881069b9e49abc8e8e5a207d82be7 Mon Sep 17 00:00:00 2001 From: Olivier Winter Date: Fri, 3 Jan 2025 17:30:42 +0000 Subject: [PATCH] add waveform files to the output list --- ibllib/pipes/ephys_tasks.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/ibllib/pipes/ephys_tasks.py b/ibllib/pipes/ephys_tasks.py index 1d29d4cec..c396fc5e0 100644 --- a/ibllib/pipes/ephys_tasks.py +++ b/ibllib/pipes/ephys_tasks.py @@ -813,7 +813,7 @@ def _run(self): spikes = alfio.load_object(probe_out_path, 'spikes', attribute=['samples', 'clusters']) clusters = alfio.load_object(probe_out_path, 'clusters', attribute=['channels']) channels = alfio.load_object(probe_out_path, 'channels') - extract_wfs_cbin( + _output_waveform_files = extract_wfs_cbin( bin_file=ap_file, output_dir=probe_out_path, spike_samples=spikes['samples'], @@ -829,6 +829,7 @@ def _run(self): preprocess_steps=["phase_shift", "bad_channel_interpolation", "butterworth", "car"], scratch_dir=self.scratch_folder_run, ) + out_files.extend(_output_waveform_files) _logger.info(f"Cleaning up temporary folder {self.scratch_folder_run}") shutil.rmtree(self.scratch_folder_run, ignore_errors=True) if self.one: @@ -852,5 +853,5 @@ def _run(self): chns = np.load(probe_out_path.joinpath('channels.localCoordinates.npy')) out = get_aligned_channels(ins[0], chns, one=self.one, save_dir=probe_out_path) out_files.extend(out) - + self.assert_expected_outputs() return out_files