diff --git a/examples/benchmarks/TFT/tft.py b/examples/benchmarks/TFT/tft.py
index 92d957d0f..a3b4fc919 100644
--- a/examples/benchmarks/TFT/tft.py
+++ b/examples/benchmarks/TFT/tft.py
@@ -232,7 +232,7 @@ class TFTModel(ModelFT):
             p90_forecast = self.data_formatter.format_predictions(output_map["p90"])
             tf.keras.backend.set_session(default_keras_session)
 
-        predict = format_score(p90_forecast, "pred", 0) # self.label_shift
+        predict = format_score(p90_forecast, "pred", 0)  # self.label_shift
         label = format_score(targets, "label", 0)
         # ===========================Predicting Process===========================
         return predict, label