From dfbb2e2b35e79f023bdfb98ca8a8e840ea4fae8d Mon Sep 17 00:00:00 2001 From: Yinon Polak Date: Wed, 8 Mar 2023 16:03:36 +0200 Subject: [PATCH] sort imports --- freqtrade/freqai/base_models/BasePyTorchModel.py | 1 + freqtrade/freqai/base_models/PyTorchModelTrainer.py | 8 ++++---- .../prediction_models/PyTorchClassifierMultiTarget.py | 9 +++------ freqtrade/freqai/prediction_models/PyTorchMLPModel.py | 5 ++--- 4 files changed, 10 insertions(+), 13 deletions(-) diff --git a/freqtrade/freqai/base_models/BasePyTorchModel.py b/freqtrade/freqai/base_models/BasePyTorchModel.py index 1074ddeea..efc36fdec 100644 --- a/freqtrade/freqai/base_models/BasePyTorchModel.py +++ b/freqtrade/freqai/base_models/BasePyTorchModel.py @@ -8,6 +8,7 @@ from pandas import DataFrame from freqtrade.freqai.data_kitchen import FreqaiDataKitchen from freqtrade.freqai.freqai_interface import IFreqaiModel + logger = logging.getLogger(__name__) diff --git a/freqtrade/freqai/base_models/PyTorchModelTrainer.py b/freqtrade/freqai/base_models/PyTorchModelTrainer.py index d02f1d896..464c5dc43 100644 --- a/freqtrade/freqai/base_models/PyTorchModelTrainer.py +++ b/freqtrade/freqai/base_models/PyTorchModelTrainer.py @@ -1,13 +1,13 @@ import logging from pathlib import Path from typing import Dict -from torch.optim import Optimizer +import pandas as pd import torch import torch.nn as nn -from torch.utils.data import DataLoader -from torch.utils.data import TensorDataset -import pandas as pd +from torch.optim import Optimizer +from torch.utils.data import DataLoader, TensorDataset + logger = logging.getLogger(__name__) diff --git a/freqtrade/freqai/prediction_models/PyTorchClassifierMultiTarget.py b/freqtrade/freqai/prediction_models/PyTorchClassifierMultiTarget.py index e58fa9cff..3623728db 100644 --- a/freqtrade/freqai/prediction_models/PyTorchClassifierMultiTarget.py +++ b/freqtrade/freqai/prediction_models/PyTorchClassifierMultiTarget.py @@ -1,19 +1,16 @@ import logging - -from typing import Any, Dict, Tuple, List -import numpy.typing as npt +from typing import Any, Dict, Tuple import numpy as np +import numpy.typing as npt import pandas as pd import torch from pandas import DataFrame from torch.nn import functional as F -from freqtrade.exceptions import OperationalException -from freqtrade.freqai.data_kitchen import FreqaiDataKitchen - from freqtrade.freqai.base_models.BasePyTorchModel import BasePyTorchModel from freqtrade.freqai.base_models.PyTorchModelTrainer import PyTorchModelTrainer +from freqtrade.freqai.data_kitchen import FreqaiDataKitchen from freqtrade.freqai.prediction_models.PyTorchMLPModel import PyTorchMLPModel diff --git a/freqtrade/freqai/prediction_models/PyTorchMLPModel.py b/freqtrade/freqai/prediction_models/PyTorchMLPModel.py index 9bbf95019..88599acb6 100644 --- a/freqtrade/freqai/prediction_models/PyTorchMLPModel.py +++ b/freqtrade/freqai/prediction_models/PyTorchMLPModel.py @@ -1,9 +1,8 @@ import logging - -import torch -import torch.nn as nn from torch import Tensor +import torch.nn as nn + logger = logging.getLogger(__name__)