From fe26e37303ad0ee0f3b4de168577040e8a3f4e29 Mon Sep 17 00:00:00 2001 From: wrongkindofdoctor <20195932+wrongkindofdoctor@users.noreply.github.com> Date: Sun, 8 Dec 2024 11:05:30 -0500 Subject: [PATCH] add logic to define start and end points for situation where desired date range is contained by xarray dataset to crop_date_range --- src/preprocessor.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/preprocessor.py b/src/preprocessor.py index e1ba1a54a..cdce8e22e 100644 --- a/src/preprocessor.py +++ b/src/preprocessor.py @@ -881,6 +881,10 @@ def crop_date_range(self, case_date_range: util.DateRange, xr_ds, time_coord) -> # dataset overlaps user-specified date range end elif date_range_cf_start < ds_start <= date_range_cf_end <= ds_end: new_xr_ds = xr_ds.sel({time_coord.name: slice(ds_start, date_range_cf_end)}) + # dataset contains all of requested date range + elif date_range_cf_start>=ds_start and date_range_cf_end<=ds_end: + new_xr_ds = xr_ds.sel({time_coord.name: slice(date_range_cf_start, date_range_cf_end)}) + return new_xr_ds def check_group_daterange(self, df: pd.DataFrame, date_range: util.DateRange,