mo global data support (#367)

Author: Sukh-P

ocf_datapipes/load/nwp/nwp.py

@@ -42,7 +42,7 @@ def __init__(
             self.open_nwp = open_icon_eu
         elif provider.lower() == "icon-global":
             self.open_nwp = open_icon_global
-        elif provider.lower() == "ecmwf":
+        elif provider.lower() in ("ecmwf", "mo_global"):  # same schema so using the same loader
             self.open_nwp = open_ifs
         elif provider.lower() == "gfs":
             self.open_nwp = open_gfs

ocf_datapipes/utils/consts.py

@@ -46,6 +46,7 @@ def __getitem__(self, key):
     "excarta",
     "merra2",
     "merra2_uk",
+    "mo_global",
 ]
 
 # ------ UKV
@@ -131,6 +132,24 @@ def __getitem__(self, key):
 UKV_STD = _to_data_array(UKV_STD)
 UKV_MEAN = _to_data_array(UKV_MEAN)
 
+# These were calculated from 200 random init times (step 0s) from the MO global data
+MO_GLOBAL_INDIA_MEAN = {
+    "temperature_sl": 298.2,
+    "wind_u_component_10m": 0.5732,
+    "wind_v_component_10m": -0.2831,
+}
+
+MO_GLOBAL_INDIA_STD = {
+    "temperature_sl": 8.473,
+    "wind_u_component_10m": 2.599,
+    "wind_v_component_10m": 2.016,
+}
+
+
+MO_GLOBAL_VARIABLE_NAMES = tuple(MO_GLOBAL_INDIA_MEAN.keys())
+MO_GLOBAL_INDIA_STD = _to_data_array(MO_GLOBAL_INDIA_STD)
+MO_GLOBAL_INDIA_MEAN = _to_data_array(MO_GLOBAL_INDIA_MEAN)
+
 
 # ------ GFS
 GFS_STD = {
@@ -355,6 +374,7 @@ def __getitem__(self, key):
     excarta=EXCARTA_VARIABLE_NAMES,
     merra2=MERRA2_VARIABLE_NAMES,
     merra2_uk=UK_MERRA2_VARIABLE_NAMES,
+    mo_global=MO_GLOBAL_VARIABLE_NAMES,
 )
 NWP_STDS = NWPStatDict(
     ukv=UKV_STD,
@@ -364,6 +384,7 @@ def __getitem__(self, key):
     excarta=EXCARTA_STD,
     merra2=MERRA2_STD,
     merra2_uk=UK_MERRA2_STD,
+    mo_global=MO_GLOBAL_INDIA_STD,
 )
 NWP_MEANS = NWPStatDict(
     ukv=UKV_MEAN,
@@ -373,6 +394,7 @@ def __getitem__(self, key):
     excarta=EXCARTA_MEAN,
     merra2=MERRA2_MEAN,
     merra2_uk=UK_MERRA2_MEAN,
+    mo_global=MO_GLOBAL_INDIA_MEAN,
 )
 
 # --------------------------- SATELLITE ------------------------------