fix: AQ Dataset

strongio · Oct 8, 2024 · 9b6e23e · 9b6e23e
1 parent c675f04
commit 9b6e23e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/torchcast/state_space/predictions.py b/torchcast/state_space/predictions.py
@@ -173,9 +173,9 @@ def with_new_start_times(self,
                              **kwargs) -> 'Predictions':
         """
         :param start_times: An array/sequence containing the start time for each group; or a single datetime to apply
-         to all groups. If the model/predictions are dateless (no dt_unit) then simply an array of indices.
+          to all groups. If the model/predictions are dateless (no dt_unit) then simply an array of indices.
         :param n_timesteps: Each group will be sliced to this many timesteps, so times is start and times + n_timesteps
-        is end.
+          is end.
         :return: A new ``Predictions`` object, with the state and measurement tensors sliced to the given times.
         """
         start_indices = self._standardize_times(times=start_times, *kwargs)

diff --git a/torchcast/utils/datasets.py b/torchcast/utils/datasets.py
@@ -30,7 +30,7 @@ def load_air_quality_data(dt_unit: str) -> 'DataFrame':
     except FileNotFoundError:
         print('loading from gh...')
         read_csv(
-            f'https://raw.githubusercontent.com/strongio/torch-kalman/5fde343674ca1fe82282477008b915794c8caaa5/examples/aq_{dt_unit}.csv',
+            f'https://raw.githubusercontent.com/strongio/torchcast/c675f04bb244a73155fbd98c110590bd735808bc/docs/examples/aq_{dt_unit}.csv',
             parse_dates=['date' if dt_unit == 'daily' else 'week']
         ).to_csv(f'/tmp/aq-{dt_unit}.csv', index=False)
     return load_air_quality_data(dt_unit)