Skip to content

Commit 3631157

Browse files
ensure top level udf output can handle geoseries
1 parent d262197 commit 3631157

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

openeo_driver/ProcessGraphDeserializer.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1450,8 +1450,11 @@ def run_udf(args: dict, env: EvalEnv):
14501450

14511451
result_collections = result_data.get_feature_collection_list()
14521452
if result_collections != None and len(result_collections) > 0:
1453-
with tempfile.NamedTemporaryFile(suffix=".json.tmp", delete=False) as temp_file:
1454-
return DriverVectorCube.from_geodataframe(data=result_collections[0].data)
1453+
geo_data = result_collections[0].data
1454+
dataframe = geo_data
1455+
if isinstance(geo_data, gpd.GeoSeries):
1456+
dataframe = gpd.GeoDataFrame(geometry=geo_data)
1457+
return DriverVectorCube.from_geodataframe(data=dataframe)
14551458
structured_result = result_data.get_structured_data_list()
14561459
if structured_result != None and len(structured_result)>0:
14571460
return JSONResult(structured_result[0].data)

0 commit comments

Comments
 (0)