fix a minor bug

yumengch · Aug 1, 2024 · 4734f5d · 4734f5d
1 parent 4eea172
commit 4734f5d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/example/prepost_processing.py b/example/prepost_processing.py
@@ -45,7 +45,7 @@ def __init__(self, model_t: model.model, pe:parallelisation.parallelisation) ->
     def get_full_ens(self, dim_p:int, dim_ens:int, ens_p:np.ndarray) -> typing.Union[np.ndarray, None]:
         """Gather total ensemble from each local processors
         """
-        if pe.npes_filter == 1: return ens_p
+        if self.pe.npes_filter == 1: return ens_p
         # get total dim
 
         ## collect full ensemble from domain decomposed ensemble
@@ -136,4 +136,4 @@ def prepostprocess(self, step:int, dim_p:int, dim_ens:int, dim_ens_p:int,
           self.preprocess(step, dim_p, dim_ens, ens_p)
         else:
           self.postprocess(step, dim_p, dim_ens, ens_p)
-        return state_p, uinv, ens_p
+        return state_p, uinv, ens_p