fix bug in test mhw

NOAA-CEFI-Portal · Sep 3, 2024 · 75ba03e · 75ba03e
1 parent 48ac32d
commit 75ba03e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/mom6/mom6_module/mom6_mhw.py b/mom6/mom6_module/mom6_mhw.py
@@ -109,6 +109,8 @@ def generate_forecast_batch(
         # anomaly used for the threshold
         ds_anom = xr.Dataset()
         ds_anom[f'{self.varname}_anom'] = dict_anom['anomaly']
+        ds_anom['lon'] = self.dataset['lon']
+        ds_anom['lat'] = self.dataset['lat']
 
         # calculate threshold
         class_forecast_quantile = ForecastQuantile(ds_anom,f'{self.varname}_anom')

diff --git a/tests/test_mom6_mhw.py b/tests/test_mom6_mhw.py
@@ -59,8 +59,8 @@ def test_MarineHeatwaveForecast_generate_forecast_single(ds_forecast):
     )
 
     assert np.abs(ds_mhw_single.mhw_prob90.sum().data-505.)==0.
-    assert np.abs(ds_mhw_single.ssta_avg.max().data-0.37654305) < 1e-6
-    assert np.abs(ds_mhw_single.mhw_mag_indentified_ens.max().data-0.37654305) < 1e-6
+    assert np.abs(ds_mhw_single.ssta_avg.max().data-0.37654305) < 1e-5
+    assert np.abs(ds_mhw_single.mhw_mag_indentified_ens.max().data-0.37654305) < 1e-5
 
 
 def test_MarineHeatwaveForecast_generate_forecast_batch(ds_forecast):