From 7bed3b4c2eba8ec0a04bdd2c1227be72d267f820 Mon Sep 17 00:00:00 2001
From: D-X-Y <280835372@qq.com>
Date: Mon, 8 Mar 2021 06:39:03 +0000
Subject: [PATCH] Fix python format by black

---
 qlib/contrib/model/pytorch_tabnet.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/qlib/contrib/model/pytorch_tabnet.py b/qlib/contrib/model/pytorch_tabnet.py
index 7e47f58dc..682e1b19f 100644
--- a/qlib/contrib/model/pytorch_tabnet.py
+++ b/qlib/contrib/model/pytorch_tabnet.py
@@ -100,7 +100,9 @@ class TabnetModel(Model):
             self.device
         )
         self.logger.info("model:\n{:}\n{:}".format(self.tabnet_model, self.tabnet_decoder))
-        self.logger.info("model size: {:.4f} MB".format(count_parameters(self.tabnet_model) + count_parameters(self.tabnet_decoder)))
+        self.logger.info(
+            "model size: {:.4f} MB".format(count_parameters(self.tabnet_model) + count_parameters(self.tabnet_decoder))
+        )
 
         if optimizer.lower() == "adam":
             self.pretrain_optimizer = optim.Adam(