more debugging output for postprocessing

henryaddison · Sep 11, 2023 · 574eccf · 574eccf
1 parent a36798f
commit 574eccf
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/ml_downscaling_emulator/bin/postprocess.py b/src/ml_downscaling_emulator/bin/postprocess.py
@@ -44,7 +44,7 @@ def filter(
     )
     os.makedirs(filtered_samples_dirpath, exist_ok=False)
 
-    for sample_filepath in samples_glob(
+    samples_filepaths_to_filter = samples_glob(
         samples_path(
             workdir,
             checkpoint=checkpoint,
@@ -53,7 +53,10 @@ def filter(
             split=split,
             ensemble_member=ensemble_member,
         )
-    ):
+    )
+
+    logger.info(f"Found for filtering: {samples_filepaths_to_filter}")
+    for sample_filepath in samples_filepaths_to_filter:
         logger.info(f"Working on {sample_filepath}")
         samples_ds = xr.open_dataset(sample_filepath)