diff --git a/freqtrade/freqai/prediction_models/PyTorchMLPModel.py b/freqtrade/freqai/prediction_models/PyTorchMLPModel.py
index 88599acb6..de9c25293 100644
--- a/freqtrade/freqai/prediction_models/PyTorchMLPModel.py
+++ b/freqtrade/freqai/prediction_models/PyTorchMLPModel.py
@@ -1,8 +1,7 @@
 import logging
 
-from torch import Tensor
 import torch.nn as nn
-
+from torch import Tensor
 
 logger = logging.getLogger(__name__)