bugfix wrt slicing datetime in xarray

MeteoSwiss · Apr 28, 2024 · df6b651 · df6b651
1 parent 32f18ed
commit df6b651
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/neural_lam/weather_dataset.py b/neural_lam/weather_dataset.py
@@ -54,7 +54,7 @@ def __init__(
         if split == "train":
             self.ds = self.ds.sel(time=slice("2015", "2019"))
         else:
-            self.ds = self.ds.sel(time=slice("2020"))
+            self.ds = self.ds.sel(time=slice("2020-01-01", "2020-12-31"))
 
         new_vars = {}
         forcings = {}