From 9bae1f766bc3d6deef450250b9656314c75590dd Mon Sep 17 00:00:00 2001 From: Sarah Jordan Date: Thu, 6 Jun 2024 10:17:51 -0500 Subject: [PATCH] update time querying --- pygeoapi/provider/xarray_edr.py | 19 +++++++------------ 1 file changed, 7 insertions(+), 12 deletions(-) diff --git a/pygeoapi/provider/xarray_edr.py b/pygeoapi/provider/xarray_edr.py index f3c3e1af8..ae6e44eaf 100644 --- a/pygeoapi/provider/xarray_edr.py +++ b/pygeoapi/provider/xarray_edr.py @@ -124,23 +124,18 @@ def position(self, **kwargs): data = self._data if self.time_field in query_params: - remaining_query = { - key: val for key, val in query_params.items() - if key != self.time_field - } if isinstance(query_params[self.time_field], slice): + remaining_query = { + key: val for key, val in query_params.items() + if key != self.time_field + } time_query = { self.time_field: query_params[self.time_field] } + data = data.sel( + time_query).sel(remaining_query, method='nearest') else: - time_query = { - self.time_field: ( - data[self.time_field].dt.date == - query_params[self.time_field] - ) - } - data = data.sel( - time_query).sel(remaining_query, method='nearest') + data = data.sel(query_params, method='nearest') else: data = data.sel(query_params, method='nearest') except KeyError: