Added subsampling

WorldCereal · Mar 20, 2024 · 9a14cee · 9a14cee
1 parent c3ee0c5
commit 9a14cee
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/scripts/extractions/extract_optical.py b/scripts/extractions/extract_optical.py
@@ -329,6 +329,7 @@ def post_job_action(
     _pipeline_log.warning(
         "Sub-sampling the job dataframe for testing. Remove this for production."
     )
+    job_df = job_df.iloc[[0]]
 
     # Setup the memory parameters for the job creator.
     create_datacube_optical = partial(

diff --git a/scripts/extractions/extract_sar.py b/scripts/extractions/extract_sar.py
@@ -356,6 +356,7 @@ def post_job_action(
     _pipeline_log.warning(
         "Sub-sampling the job dataframe for testing. Remove this for production."
     )
+    job_df = job_df.iloc[[0]]
 
     # Setup the memory parameters for the job creator.
     create_datacube_sar = partial(