[JTH] little edit to homogenize new base_wrappers posprocess_cases function

Author: tausiaj

bluemath_tk/wrappers/lisflood/__init__.py

@@ -341,6 +341,7 @@ def postprocess_case(
         self,
         case_num: int,
         case_dir: str,
+        case_context: dict,
         output_vars: List[str] = None,
         overwrite_output: bool = True,
         overwrite_output_postprocessed: bool = True,
@@ -356,6 +357,8 @@ def postprocess_case(
             The case number.
         case_dir : str
             The case directory.
+        case_context : dict
+            The case context.
         output_vars : list, optional
             The output variables to postprocess. Default is None.
         overwrite_output : bool, optional

bluemath_tk/wrappers/lisflood/lisflood_wrapper.py

@@ -5,9 +5,8 @@
 import numpy as np
 import pandas as pd
 import xarray as xr
-
-from wavespectra.construct.frequency import jonswap
 from wavespectra.construct.direction import cartwright
+from wavespectra.construct.frequency import jonswap
 
 from .._base_wrappers import BaseModelWrapper
 
@@ -67,7 +66,7 @@ def __init__(
         self.set_logger_name(
             name=self.__class__.__name__, level="DEBUG" if debug else "INFO"
         )
-    
+
     def create_vardens(self, ds):
         t = ""
 
@@ -89,9 +88,9 @@ def create_vardens(self, ds):
                     var = 0.0
                 t += "{0}\t".format(var)
             t += "\n"
-        
+
         return t
-    
+
     def build_case(
         self,
         case_context: dict,
@@ -131,7 +130,7 @@ def build_case(
                 dm=case_context["Dir"],
                 dspr=case_context["SPR"],
             )
-            efth = ef * gth  
+            efth = ef * gth
 
             spectrum = xr.Dataset(
                 {
@@ -146,12 +145,11 @@ def build_case(
                 },
             ).sortby(["freq", "dir"])
 
-            spectrum['dir'] = (270 - (spectrum['dir'])) % 360
+            spectrum["dir"] = (270 - (spectrum["dir"])) % 360
             spec = self.create_vardens(spectrum)
             with open(f"{case_dir}/vardens.txt", "w") as f:
                 f.write(spec)
 
-
     def _get_average_var(self, case_nc: xr.Dataset, var: str) -> np.ndarray:
         """
         Get the average value of a variable except for the first hour of the simulation
@@ -241,6 +239,7 @@ def postprocess_case(
         self,
         case_num: int,
         case_dir: str,
+        case_context: dict,
         output_vars: List[str] = None,
         overwrite_output: bool = True,
     ) -> xr.Dataset:
@@ -253,6 +252,8 @@ def postprocess_case(
             The case number.
         case_dir : str
             The case directory.
+        case_context : dict
+            The case context.
         output_vars : list, optional
             The output variables to postprocess. Default is None.
         overwrite_output : bool, optional