revert simulation dim names

ecmwf · Mar 26, 2024 · 763ccb4 · 763ccb4
1 parent a8b28d8
commit 763ccb4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 7 deletions.
diff --git a/hat/extract_simulation_timeseries.py b/hat/extract_simulation_timeseries.py
@@ -26,7 +26,7 @@ def extract_timeseries(
     config = valid_custom_config(config)
 
     # infer coordinates for all grids from first the grid
-    coords = latlon_coords(simulations_da, names=config.get("station_coordinates", []))
+    coords = latlon_coords(simulations_da)
 
     # numpy array of station locations
     station_mask = geopoints_to_array(stations, coords)

diff --git a/hat/geo.py b/hat/geo.py
@@ -347,14 +347,10 @@ def get_latlon_keys(ds):
     return lat_key, lon_key
 
 
-def latlon_coords(ds, names: list = []):
+def latlon_coords(ds):
     """Latitude and longitude coordinates of an xarray"""
 
-    if not names:
-        lat_key, lon_key = get_latlon_keys(ds)
-    else:
-        assert len(names) == 2, "Must provide two names for lat and lon"
-    lat_key, lon_key = names
+    lat_key, lon_key = get_latlon_keys(ds)
     lat_coords = ds.coords.get(lat_key).data
     lon_coords = ds.coords.get(lon_key).data
     coords = {"x": lon_coords, "y": lat_coords}