From f930ff53854fdec0d308b06862fcb9418ddbc66c Mon Sep 17 00:00:00 2001 From: Peter Dudfield <34686298+peterdudfield@users.noreply.github.com> Date: Mon, 4 Nov 2024 16:18:54 +0000 Subject: [PATCH] Update ocf_data_sampler/torch_datasets/site.py Co-authored-by: Alexandra Udaltsova <43303448+AUdaltsova@users.noreply.github.com> --- ocf_data_sampler/torch_datasets/site.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ocf_data_sampler/torch_datasets/site.py b/ocf_data_sampler/torch_datasets/site.py index 5483a22..0cce14b 100644 --- a/ocf_data_sampler/torch_datasets/site.py +++ b/ocf_data_sampler/torch_datasets/site.py @@ -66,7 +66,7 @@ def find_valid_t0_and_site_ids( # The last step of the forecast is lost if we have to diff channels if len(nwp_config.nwp_accum_channels) > 0: - end_buffer = pd.to_timedelta(nwp_config.time_resolution_minutes) + end_buffer = minutes(nwp_config.time_resolution_minutes) else: end_buffer =minutes(0)