From e58496124996797355ae4aed528fcc02f2dcff9c Mon Sep 17 00:00:00 2001
From: gorold <woogerald@yahoo.com.sg>
Date: Mon, 5 Sep 2022 22:42:24 +0800
Subject: [PATCH] minor refactor

---
 exp/exp_main.py                                          | 6 +++---
 models/__init__.py                                       | 1 +
 models/{ETSformer => etsformer}/__init__.py              | 0
 models/{ETSformer => etsformer}/decoder.py               | 0
 models/{ETSformer => etsformer}/encoder.py               | 0
 models/{ETSformer => etsformer}/exponential_smoothing.py | 0
 models/{ETSformer => etsformer}/model.py                 | 0
 models/{ETSformer => etsformer}/modules.py               | 0
 8 files changed, 4 insertions(+), 3 deletions(-)
 rename models/{ETSformer => etsformer}/__init__.py (100%)
 rename models/{ETSformer => etsformer}/decoder.py (100%)
 rename models/{ETSformer => etsformer}/encoder.py (100%)
 rename models/{ETSformer => etsformer}/exponential_smoothing.py (100%)
 rename models/{ETSformer => etsformer}/model.py (100%)
 rename models/{ETSformer => etsformer}/modules.py (100%)

diff --git a/exp/exp_main.py b/exp/exp_main.py
index db79626..839f93e 100644
--- a/exp/exp_main.py
+++ b/exp/exp_main.py
@@ -1,4 +1,4 @@
-from models.ETSformer.model import ETSformer
+from models import ETSformer
 from data_provider.data_factory import data_provider
 from exp.exp_basic import Exp_Basic
 from utils.tools import EarlyStopping, adjust_learning_rate
@@ -210,8 +210,8 @@ class Exp_Main(Exp_Basic):
                 outputs = outputs.detach().cpu().numpy()
                 batch_y = batch_y.detach().cpu().numpy()
 
-                pred = outputs  # outputs.detach().cpu().numpy()  # .squeeze()
-                true = batch_y  # batch_y.detach().cpu().numpy()  # .squeeze()
+                pred = outputs
+                true = batch_y
 
                 preds.append(pred)
                 trues.append(true)
diff --git a/models/__init__.py b/models/__init__.py
index e69de29..adbf454 100644
--- a/models/__init__.py
+++ b/models/__init__.py
@@ -0,0 +1 @@
+from .etsformer.model import ETSformer
diff --git a/models/ETSformer/__init__.py b/models/etsformer/__init__.py
similarity index 100%
rename from models/ETSformer/__init__.py
rename to models/etsformer/__init__.py
diff --git a/models/ETSformer/decoder.py b/models/etsformer/decoder.py
similarity index 100%
rename from models/ETSformer/decoder.py
rename to models/etsformer/decoder.py
diff --git a/models/ETSformer/encoder.py b/models/etsformer/encoder.py
similarity index 100%
rename from models/ETSformer/encoder.py
rename to models/etsformer/encoder.py
diff --git a/models/ETSformer/exponential_smoothing.py b/models/etsformer/exponential_smoothing.py
similarity index 100%
rename from models/ETSformer/exponential_smoothing.py
rename to models/etsformer/exponential_smoothing.py
diff --git a/models/ETSformer/model.py b/models/etsformer/model.py
similarity index 100%
rename from models/ETSformer/model.py
rename to models/etsformer/model.py
diff --git a/models/ETSformer/modules.py b/models/etsformer/modules.py
similarity index 100%
rename from models/ETSformer/modules.py
rename to models/etsformer/modules.py