test block has my fucking number

NCAR · Nov 8, 2024 · 6fb3191 · 6fb3191
1 parent a951c1b
commit 6fb3191
Showing 1 changed file with 5 additions and 3 deletions.
diff --git 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):
         # ------------------------------------------------------------------------------ #