fix lookback_len bug

2026-06-27 16:28:35 +08:00 · 2022-07-31 16:13:56 +08:00
parent 3e73ce5486
commit 9f6ccba9a3
1 changed files with 1 additions and 1 deletions
@@ -49,7 +49,7 @@ class ForecastDataset(Dataset):
            f"only 'lookback_len' xor 'lookback_mult' should be specified"

        self.flag = flag
-        self.lookback_len = lookback_len or int(horizon_len * lookback_mult)
+        sself.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