force a shuffle after scl mask, allowing caching

Open-EO/openeo-geopyspark-driver#986
Open-EO · Jan 8, 2025 · 9acdd8a · 9acdd8a
1 parent 37dcebd
commit 9acdd8a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/openeo-geotrellis/src/main/scala/org/openeo/geotrellis/OpenEOProcesses.scala b/openeo-geotrellis/src/main/scala/org/openeo/geotrellis/OpenEOProcesses.scala
@@ -1259,7 +1259,7 @@ class OpenEOProcesses extends Serializable {
       MultibandTile(filter.createMask(tile.tile).crop(originalBounds))
     })
     val updatedMetadata = datacube.metadata.copy(cellType = BitCellType)
-    ContextRDD(mask, updatedMetadata)
+    ContextRDD(new ShuffledRDD[SpaceTimeKey, MultibandTile,MultibandTile](mask,mask.partitioner.get), updatedMetadata)
   }
 
   def mergeTiles(tiles: MultibandTileLayerRDD[SpaceTimeKey]): MultibandTileLayerRDD[SpaceTimeKey] = {