Merge branch 'research' into feature_verification

mllam · Feb 5, 2025 · 6b7bcbd · 6b7bcbd
2 parents 75d4ddd + ba9df3a
commit 6b7bcbd
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/neural_lam/weather_dataset.py b/neural_lam/weather_dataset.py
@@ -157,12 +157,12 @@ def __init__(
                 "Created WeatherDataset for time interval "
                 f"{time_slice.start} - {time_slice.stop}"
             )
-            # Subset state, forcing
-            # Do not need to subset boundary forcing, as that is handled below
-            # with crop_time_if_needed
-
+            # Subset state, forcing, boundary_forcing
             self.da_state = self.da_state.sel(time=time_slice)
             self.da_forcing = self.da_forcing.sel(time=time_slice)
+            self.da_boundary_forcing = self.da_boundary_forcing.sel(
+                time=time_slice
+            )
 
         # check that with the provided data-arrays and ar_steps that we have a
         # non-zero amount of samples