Update ocf_data_sampler/torch_datasets/site.py

Co-authored-by: Alexandra Udaltsova <[email protected]>
openclimatefix · Nov 4, 2024 · f930ff5 · f930ff5
1 parent 972bb0f
commit f930ff5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)