Updated private functions to public in gfmap and removed unnecessary …

…featureflags
WorldCereal · May 22, 2024 · 2fb819f · 2fb819f
1 parent 49e98ab
commit 2fb819f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 13 deletions.
diff --git a/scripts/extractions/extract_sar.py b/scripts/extractions/extract_sar.py
@@ -22,8 +22,8 @@
 from openeo_gfmap.fetching.s1 import build_sentinel1_grd_extractor
 from openeo_gfmap.manager.job_manager import GFMAPJobManager
 from openeo_gfmap.manager.job_splitters import (
-    _append_h3_index,
-    _load_s2_grid,
+    append_h3_index,
+    load_s2_grid,
     split_job_s2grid,
 )
 from shapely.geometry import Point
@@ -346,7 +346,7 @@ def post_job_action(
     pipeline_log.info("Loading input dataframe from %s.", args.input_df)
 
     input_df = gpd.read_file(args.input_df)
-    input_df = _append_h3_index(input_df)
+    input_df = append_h3_index(input_df)
 
     split_dfs = split_job_s2grid(input_df, max_points=args.max_locations)
     split_dfs = [df for df in split_dfs if df.extract.any()]
@@ -368,7 +368,7 @@ def post_job_action(
     # Setup the s2 grid for the output path generation function
     generate_output_path = partial(
         generate_output_path,
-        s2_grid=_load_s2_grid(),
+        s2_grid=load_s2_grid(),
     )
 
     manager = GFMAPJobManager(

diff --git a/src/worldcereal/openeo/preprocessing.py b/src/worldcereal/openeo/preprocessing.py
@@ -116,15 +116,6 @@ def raw_datacube_S2(
             "eo:cloud_cover": lambda val: val <= 95.0,
         },
     }
-    # if (backend_context.backend is Backend.CDSE_STAGING) and (
-    #     fetch_type == FetchType.POLYGON
-    # ):  # At the moment only available in CDSE staging
-    #     extraction_parameters["update_arguments"] = {
-    #         "featureflags": {
-    #             "tile_size": 128,
-    #             "load_per_product": True,
-    #         }
-    #     }
     if additional_masks:
         extraction_parameters["pre_merge"] = additional_masks
     if filter_tile: