rename AggregatePolygonSpatialResult.get_data() to _get_geodataframe() …

…Open-EO/openeo-geopyspark-driver#663
Open-EO · Mar 6, 2024 · 77644dc · 77644dc
1 parent b01b5ef
commit 77644dc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/openeo_driver/save_result.py b/openeo_driver/save_result.py
@@ -771,7 +771,7 @@ def fit_class_random_forest(
         # TODO: this method belongs eventually under DriverVectorCube
         raise NotImplementedError
 
-    def get_data(self) -> gpd.GeoDataFrame:
+    def _get_geodataframe(self) -> gpd.GeoDataFrame:
         if isinstance(self._regions, DriverVectorCube):
             gdf = gpd.GeoDataFrame(geometry=self._regions.get_geometries())
         elif isinstance(self._regions, GeometryCollection):
@@ -788,7 +788,7 @@ def get_data(self) -> gpd.GeoDataFrame:
         return gdf.drop(columns=["feature_index"])
 
     def to_driver_vector_cube(self) -> DriverVectorCube:
-        gdf = self.get_data()
+        gdf = self._get_geodataframe()
         return DriverVectorCube.from_geodataframe(gdf, dimension_name="bands")