From c90aa239fea1e7e450d783468b5d3b193c4c62b8 Mon Sep 17 00:00:00 2001 From: Yingkai Sha Date: Mon, 4 Nov 2024 21:03:30 -0700 Subject: [PATCH] bugfix on pytest --- credit/postblock.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/credit/postblock.py b/credit/postblock.py index ea73c80..f841bea 100644 --- 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,