We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent d11fa17 commit ad97c0fCopy full SHA for ad97c0f
1 file changed
src/spikeinterface/core/motion.py
@@ -1,5 +1,6 @@
1
import json
2
from pathlib import Path
3
+import shutil
4
5
import numpy as np
6
import spikeinterface
@@ -171,9 +172,14 @@ def from_dict(d):
171
172
interpolation_method=d["interpolation_method"],
173
)
174
- def save(self, folder):
175
+ def save(self, folder, overwrite=False):
176
folder = Path(folder)
- 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)
183
184
info_file = folder / f"spikeinterface_info.json"
185
info = dict(
0 commit comments