Update to main save/load methods

Signed-off-by: Harm Matthias Harms <[email protected]>
kedro-org · Oct 7, 2024 · b789b01 · b789b01
1 parent 29929d6
commit b789b01
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/kedro-datasets/kedro_datasets/geopandas/generic_dataset.py b/kedro-datasets/kedro_datasets/geopandas/generic_dataset.py
@@ -149,7 +149,7 @@ def _ensure_file_system_target(self) -> None:
                 f"does not support a filepath target/source."
             )
 
-    def _load(self) -> gpd.GeoDataFrame | dict[str, gpd.GeoDataFrame]:
+    def load(self) -> gpd.GeoDataFrame | dict[str, gpd.GeoDataFrame]:
         self._ensure_file_system_target()
 
         load_path = get_filepath_str(self._get_load_path(), self._protocol)
@@ -163,7 +163,7 @@ def _load(self) -> gpd.GeoDataFrame | dict[str, gpd.GeoDataFrame]:
             "https://geopandas.org/en/stable/docs/reference/io.html"
         )
 
-    def _save(self, data: gpd.GeoDataFrame) -> None:
+    def save(self, data: gpd.GeoDataFrame) -> None:
         self._ensure_file_system_target()
 
         save_path = get_filepath_str(self._get_save_path(), self._protocol)