Skip to content

Commit bffd5e9

Browse files
authored
[ENH] Add overwrite option to transform_images (#509)
* Add overwrite option to transform_images. * Apply suggestions from code review
1 parent ed4abc4 commit bffd5e9

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

nimare/transforms.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@
1818
LGR = logging.getLogger(__name__)
1919

2020

21-
def transform_images(images_df, target, masker, metadata_df=None, out_dir=None):
21+
def transform_images(images_df, target, masker, metadata_df=None, out_dir=None, overwrite=False):
2222
"""Generate images of a given type from other image types and write out to files.
2323
2424
Parameters
@@ -37,6 +37,8 @@ def transform_images(images_df, target, masker, metadata_df=None, out_dir=None):
3737
out_dir : :obj:`str` or :obj:`None`, optional
3838
Path to output directory. If None, use folder containing first image
3939
for each study in ``images_df``.
40+
overwrite : :obj:`bool`, optional
41+
Whether to overwrite existing files or not. Default is False.
4042
4143
Returns
4244
-------
@@ -84,7 +86,11 @@ def transform_images(images_df, target, masker, metadata_df=None, out_dir=None):
8486
# Get converted data
8587
img = resolve_transforms(target, available_data, new_masker)
8688
if img is not None:
87-
img.to_filename(new_file)
89+
if overwrite or not op.isfile(new_file):
90+
img.to_filename(new_file)
91+
else:
92+
LGR.debug("Image already exists. Not overwriting.")
93+
8894
images_df.loc[images_df["id"] == id_, target] = new_file
8995
else:
9096
images_df.loc[images_df["id"] == id_, target] = None

0 commit comments

Comments
 (0)