avoid accessing private attribute Open-EO/openeo-geopyspark-driver#663

Open-EO · Mar 6, 2024 · b01b5ef · b01b5ef
1 parent d367a3c
commit b01b5ef
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/openeo_driver/save_result.py b/openeo_driver/save_result.py
@@ -773,7 +773,7 @@ def fit_class_random_forest(
 
     def get_data(self) -> gpd.GeoDataFrame:
         if isinstance(self._regions, DriverVectorCube):
-            gdf = self._regions._geometries
+            gdf = gpd.GeoDataFrame(geometry=self._regions.get_geometries())
         elif isinstance(self._regions, GeometryCollection):
             gdf = gpd.GeoDataFrame(geometry=list(self._regions.geoms))
         elif isinstance(self._regions, BaseGeometry):