From 1ac5b2c8dc71987b7319845a11e1bec5f6d3d54b Mon Sep 17 00:00:00 2001 From: Gabriel Bellido Date: Wed, 19 Nov 2025 12:28:55 +0100 Subject: [PATCH] [GBP] add case_context to base_wrappers class postrocess method --- bluemath_tk/wrappers/_base_wrappers.py | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/bluemath_tk/wrappers/_base_wrappers.py b/bluemath_tk/wrappers/_base_wrappers.py index 9383d48..151f265 100644 --- a/bluemath_tk/wrappers/_base_wrappers.py +++ b/bluemath_tk/wrappers/_base_wrappers.py @@ -838,15 +838,24 @@ def postprocess_cases( cases_dir_to_postprocess = [ self.cases_dirs[case] for case in cases_to_postprocess ] + cases_context_to_postprocess = [ + self.cases_context[case] for case in cases_to_postprocess + ] else: cases_to_postprocess = list(range(len(self.cases_dirs))) cases_dir_to_postprocess = copy.deepcopy(self.cases_dirs) + cases_context_to_postprocess = copy.deepcopy(self.cases_context) postprocessed_files = [] - for case_num, case_dir in zip(cases_to_postprocess, cases_dir_to_postprocess): + for case_num, case_dir, case_context in zip( + cases_to_postprocess, cases_dir_to_postprocess, cases_context_to_postprocess + ): try: postprocessed_file = self.postprocess_case( - case_num=case_num, case_dir=case_dir, **kwargs + case_num=case_num, + case_dir=case_dir, + case_context=case_context, + **kwargs, ) postprocessed_files.append(postprocessed_file) except Exception as e: