From 6b4156ab9fb5dbe3062dbe4bc347fc6ddcb8cf86 Mon Sep 17 00:00:00 2001 From: Dong Zhou Date: Fri, 27 Nov 2020 08:58:44 +0800 Subject: [PATCH] black format --- qlib/contrib/model/pytorch_hats.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/qlib/contrib/model/pytorch_hats.py b/qlib/contrib/model/pytorch_hats.py index 1eff35203..a0da88dbf 100644 --- a/qlib/contrib/model/pytorch_hats.py +++ b/qlib/contrib/model/pytorch_hats.py @@ -180,7 +180,7 @@ class HATS(Model): def train_epoch(self, x_train, y_train): x_train_values = x_train.values - y_train_values = np.squeeze(y_train.values) + y_train_values = np.squeeze(y_train.values) self.HATS_model.train()