proper metadata dimensions in raster_to_vector

#303
Open-EO · Aug 12, 2024 · 22f89cf · 22f89cf
1 parent 4cf297c
commit 22f89cf
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/openeo_driver/dry_run.py b/openeo_driver/dry_run.py
@@ -619,7 +619,13 @@ def _normalize_geometry(self, geometries) -> Tuple[Union[DriverVectorCube, Delay
         return geometries, bbox
 
     def raster_to_vector(self):
-        return self._process(operation="raster_to_vector", arguments={})
+        dimensions = [SpatialDimension(name=DriverVectorCube.DIM_GEOMETRY,extent=self.metadata.extent)]
+        if(self.metadata.has_temporal_dimension()):
+            dimensions.append(self.metadata.temporal_dimension)
+        if(self.metadata.has_band_dimension()):
+            dimensions.append(self.metadata.band_dimension)
+
+        return self._process(operation="raster_to_vector", arguments={},metadata=CollectionMetadata(metadata={}, dimensions=dimensions))
 
 
     def resample_cube_spatial(self, target: 'DryRunDataCube', method: str = 'near') -> 'DryRunDataCube':