diff --git a/minigrida/loader/coord_tiles.py b/minigrida/loader/coord_tiles.py index c137d98..dfcdb8c 100644 --- a/minigrida/loader/coord_tiles.py +++ b/minigrida/loader/coord_tiles.py @@ -25,7 +25,8 @@ def run(rasters_path, rasters_name, gt_suffix='gt.tif'): for gtn in gt_names: gt += [load_tif(gtn)] - rasters += [{Path(n).stem: load_tif(gtn.as_posix().strip(gt_suffix) + n) + rasters += [{Path(n).stem: + load_tif(gtn.as_posix().replace(gt_suffix, '') + n) for n in rasters_name}] return gt, rasters