From 9f6ccba9a3dd559088f52daec6f2798a2741c203 Mon Sep 17 00:00:00 2001
From: gorold <woogerald@yahoo.com.sg>
Date: Sun, 31 Jul 2022 16:13:56 +0800
Subject: [PATCH] fix lookback_len bug

---
 data/datasets.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/data/datasets.py b/data/datasets.py
index 58a1508..118ef6f 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
-        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