@@ -76,9 +76,9 @@ def _(ArgumentParser, Path):
7676def _ (grib_dir , init_time , lead_time , load_state_from_grib , param ):
7777 # load grib file
7878 grib_file = grib_dir / f"{ init_time } _{ lead_time } .grib"
79- if param == "10sp " :
79+ if param == "10si " :
8080 paramlist = ["10u" , "10v" ]
81- elif param == "sp " :
81+ elif param == "si " :
8282 paramlist = ["u" , "v" ]
8383 else :
8484 paramlist = [param ]
@@ -159,21 +159,21 @@ def _m_to_mm(arr):
159159 def preprocess_field (param : str , state : dict ):
160160 """
161161 - Temperatures (2t, 2d, t, d): K -> °C
162- - Wind speed at 10m (10sp ): m/s -> kn, sqrt(10u^2 + 10v^2)
163- - Wind speed (sp ): m/s -> kn, sqrt(u^2 + v^2)
162+ - Wind speed at 10m (10si ): m/s -> kn, sqrt(10u^2 + 10v^2)
163+ - Wind speed (si ): m/s -> kn, sqrt(u^2 + v^2)
164164 Returns: (field_array, units_override or None)
165165 """
166166 fields = state ["fields" ]
167167 # temperature variables
168168 if param in ("2t" , "2d" , "t" , "d" ):
169169 return _k_to_c (fields [param ]), "°C"
170170 # 10m wind speed (allow legacy 'uv' alias)
171- if param == "10sp " :
171+ if param == "10si " :
172172 u = fields ["10u" ]
173173 v = fields ["10v" ]
174174 return np .sqrt (u ** 2 + v ** 2 ), "m/s"
175175 # wind speed from standard-level components
176- if param == "sp " :
176+ if param == "si " :
177177 u = fields ["u" ]
178178 v = fields ["v" ]
179179 return np .sqrt (u ** 2 + v ** 2 ), "m/s"
0 commit comments