new observations after conservative regridding and groupby week

scoring/scoring_script.py

@@ -6,7 +6,7 @@ import pandas as pd
 import argparse
 from pathlib import Path
 
-def assert_predictions_2020(preds_test, exclude='weekofyear'):
+def assert_predictions_2020(preds_test, exclude='week'):
     """Check the variables, coordinates and dimensions of 2020 predictions."""
     from xarray.testing import assert_equal # doesnt care about attrs but checks coords
     if isinstance(exclude, str):
@@ -21,9 +21,9 @@ def assert_predictions_2020(preds_test, exclude='weekofyear'):
         assert 't2m' in preds_test.data_vars
     
     ## coords
-    # ignore weekofyear coord if not dim
-    if 'weekofyear' in exclude and 'weekofyear' in preds_test.coords and 'weekofyear' not in preds_test.dims:
-        preds_test = preds_test.drop('weekofyear')
+    # ignore week coord if not dim
+    if 'week' in exclude and 'week' in preds_test.coords and 'week' not in preds_test.dims:
+        preds_test = preds_test.drop('week')
     
     # forecast_time
     if 'forecast_time' not in exclude: