From 4734f5d817917fc8f8ab4b256aa42987795a0845 Mon Sep 17 00:00:00 2001 From: Yumeng Chen Date: Thu, 1 Aug 2024 10:24:54 +0000 Subject: [PATCH] fix a minor bug --- example/prepost_processing.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/example/prepost_processing.py b/example/prepost_processing.py index ed32e5ee..69e11ca3 100644 --- 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 \ No newline at end of file + return state_p, uinv, ens_p