Update neural_lam/weather_dataset.py

Co-authored-by: Joel Oskarsson <[email protected]>
mllam · Sep 23, 2024 · 2b45b5a · 2b45b5a
1 parent 7dfd570
commit 2b45b5a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/neural_lam/weather_dataset.py b/neural_lam/weather_dataset.py
@@ -116,7 +116,8 @@ def _sample_time(self, da, idx, n_steps: int, n_timesteps_offset: int = 0):
             The index of the time step to start the sample from.
         n_steps : int
             The number of time steps to include in the sample.
-
+        n_timestep_offset : int
+            A number of timesteps to use as offset from the start time of the slice
         """
         # selecting the time slice
         if self.datastore.is_forecast: