ENH: Add overwrite

larsoner · Sep 12, 2024 · d007f32 · d007f32
1 parent de4ebbf
commit d007f32
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/mnefun/_ssp.py b/mnefun/_ssp.py
@@ -388,15 +388,21 @@ def do_preprocessing_combined(p, subjects, run_indices):
             print('  obtained %d epochs from %d events.' % (len(ecg_epochs),
                                                             len(ecg_events)))
             if len(ecg_epochs) >= 20:
-                write_events(ecg_eve, ecg_epochs.events)
+                kwargs = dict()
+                if "overwrite" in get_args(write_events):
+                    kwargs["overwrite"] = True
+                write_events(ecg_eve, ecg_epochs.events, **kwargs)
                 ecg_epochs.save(ecg_epo, **_get_epo_kwargs())
                 desc_prefix = 'ECG-%s-%s' % tuple(ecg_t_lims)
                 pr = compute_proj_wrap(
                     ecg_epochs, p.proj_ave, n_grad=proj_nums[0][0],
                     n_mag=proj_nums[0][1], n_eeg=proj_nums[0][2],
                     desc_prefix=desc_prefix, **proj_kwargs)
                 assert len(pr) == np.sum(proj_nums[0][::p_sl])
-                write_proj(ecg_proj, pr)
+                kwargs = dict()
+                if "overwrite" in get_args(write_events):
+                    kwargs["overwrite"] = True
+                write_proj(ecg_proj, pr, **kwargs)
                 projs.extend(pr)
             else:
                 _raise_bad_epochs(raw, ecg_epochs, ecg_events, 'ECG',