Handling case where mask raster was just created. RE:natcap#366

phargogh · Jan 31, 2024 · ef70242 · ef70242
1 parent b760933
commit ef70242
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/pygeoprocessing/geoprocessing.py b/src/pygeoprocessing/geoprocessing.py
@@ -1167,7 +1167,10 @@ def align_and_resize_raster_stack(
         if 'mask_raster_path' in vector_mask_options:
             mask_raster_path = vector_mask_options['mask_raster_path']
         else:
+            # Add the mask raster path ot the vector mask options to force
+            # warp_raster to use the existing raster mask.
             mask_raster_path = os.path.join(temp_working_dir, 'mask.tif')
+            vector_mask_options['mask_raster_path'] = mask_raster_path
         new_raster_from_base(
             warped_vrt, mask_raster_path, gdal.GDT_Byte, [0], [0])