Skip to content

Commit ad97c0f

Browse files
authored
Allow overwrite argument when saving motion object (#4464)
1 parent d11fa17 commit ad97c0f

1 file changed

Lines changed: 8 additions & 2 deletions

File tree

src/spikeinterface/core/motion.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
import json
22
from pathlib import Path
3+
import shutil
34

45
import numpy as np
56
import spikeinterface
@@ -171,9 +172,14 @@ def from_dict(d):
171172
interpolation_method=d["interpolation_method"],
172173
)
173174

174-
def save(self, folder):
175+
def save(self, folder, overwrite=False):
175176
folder = Path(folder)
176-
folder.mkdir(exist_ok=False, parents=True)
177+
if folder.is_dir():
178+
if overwrite:
179+
shutil.rmtree(folder)
180+
else:
181+
raise FileExistsError(f"Folder {folder} already exists. Use `overwrite=True` to overwrite.")
182+
folder.mkdir(exist_ok=True, parents=True)
177183

178184
info_file = folder / f"spikeinterface_info.json"
179185
info = dict(

0 commit comments

Comments
 (0)