check for to_driver_vector_cube attribute in vector_to_raster Open-EO…

…/openeo-geopyspark-driver#663
Open-EO · Mar 13, 2024 · b52bd26 · b52bd26
1 parent faeb345
commit b52bd26
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/openeo_driver/ProcessGraphDeserializer.py b/openeo_driver/ProcessGraphDeserializer.py
@@ -1752,7 +1752,8 @@ def raster_to_vector(args: Dict, env: EvalEnv):
 def vector_to_raster(args: dict, env: EvalEnv) -> DriverDataCube:
     input_vector_cube = extract_arg(args, "data")
     target_data_cube = extract_arg(args, "target_data_cube")  # TODO: Rename to 'target'.
-    if not isinstance(input_vector_cube, DriverVectorCube):
+    # TODO: to_driver_vector_cube is temporary. Remove it when vector cube is fully supported.
+    if not isinstance(input_vector_cube, DriverVectorCube) and not hasattr(input_vector_cube, "to_driver_vector_cube"):
         raise ProcessParameterInvalidException(
             parameter="data",
             process="vector_to_raster",