fixup! Improve auto-adding of save_result (#623, #401, #583, #391)

Open-EO · Sep 25, 2024 · a95d73f · a95d73f
1 parent 5aa28fb
commit a95d73f
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/openeo/rest/datacube.py b/openeo/rest/datacube.py
@@ -2095,9 +2095,6 @@ def save_result(
             }
         )
 
-    def _get_save_result_nodes(self) -> List[PGNode]:
-        return [n for n in self.result_node().walk_nodes() if n.process_id == "save_result"]
-
     def _ensure_save_result(
         self,
         *,
@@ -2116,7 +2113,7 @@ def _ensure_save_result(
         :return:
         """
         # TODO #401 Unify with VectorCube._ensure_save_result and move to generic data cube parent class (not only for raster cubes, but also vector cubes)
-        save_result_nodes = self._get_save_result_nodes()
+        save_result_nodes = [n for n in self.result_node().walk_nodes() if n.process_id == "save_result"]
 
         cube = self
         if not save_result_nodes: