diff --git a/credit/postblock.py b/credit/postblock.py index a40af71..5a47e99 100644 --- a/credit/postblock.py +++ b/credit/postblock.py @@ -224,6 +224,8 @@ def __init__(self, post_conf): lat2d, p_level, midpoint=self.flag_midpoint) + + self.requires_scaling = post_conf['requires_scaling'] # -------------------------------------------------------------------------- # self.ind_fix = self.N_levels - int(post_conf['global_mass_fixer']['fix_level_num']) + 1 @@ -246,7 +248,7 @@ def __init__(self, post_conf): else: self.state_trans = None - self.requires_scaling = post_conf['requires_scaling'] + def forward(self, x): # ------------------------------------------------------------------------------ # @@ -449,7 +451,8 @@ def __init__(self, post_conf): lat2d, p_level, midpoint=self.flag_midpoint) - + + self.requires_scaling = post_conf['requires_scaling'] self.N_seconds = int(post_conf['data']['lead_time_periods']) * 3600 # ------------------------------------------------------------------------------------ # @@ -467,7 +470,6 @@ def __init__(self, post_conf): else: self.state_trans = None - self.requires_scaling = post_conf['requires_scaling'] def forward(self, x): # ------------------------------------------------------------------------------ #