bugfix on pytest

NCAR · Nov 5, 2024 · c90aa23 · c90aa23
1 parent 92d34d9
commit c90aa23
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/credit/postblock.py b/credit/postblock.py
@@ -381,7 +381,7 @@ def __init__(self, post_conf):
 
             p_level_demo = torch.from_numpy(np.array([100, 30000, 50000, 70000, 80000, 90000, 100000]))
             self.flag_sigma_level = False
-            self.flag_midpoint = post_conf['global_mass_fixer']['midpoint']
+            self.flag_midpoint = post_conf['global_water_fixer']['midpoint']
             self.core_compute = physics_pressure_level(lon_demo, lat_demo, p_level_demo, 
                                                        midpoint=self.flag_midpoint)
             self.N_levels = len(p_level_demo)
@@ -577,7 +577,7 @@ def __init__(self, post_conf):
                 np.array([100, 30000, 50000, 70000, 80000, 90000, 100000])
             )
             self.flag_sigma_level = False
-            self.flag_midpoint = post_conf['global_mass_fixer']['midpoint']
+            self.flag_midpoint = post_conf['global_energy_fixer']['midpoint']
             self.core_compute = physics_pressure_level(
                 lon_demo,
                 lat_demo,