diff --git a/data/datasets.py b/data/datasets.py index 118ef6f..14b9901 100644 --- a/data/datasets.py +++ b/data/datasets.py @@ -49,7 +49,7 @@ class ForecastDataset(Dataset): f"only 'lookback_len' xor 'lookback_mult' should be specified" self.flag = flag - sself.lookback_len = int(horizon_len * lookback_mult) if lookback_mult is not None else lookback_len + self.lookback_len = int(horizon_len * lookback_mult) if lookback_mult is not None else lookback_len self.lookback_aux_len = lookback_aux_len self.horizon_len = horizon_len self.scale = scale