diff --git a/freqtrade/freqai/data_drawer.py b/freqtrade/freqai/data_drawer.py index e1fc28d92..89708fd3e 100644 --- a/freqtrade/freqai/data_drawer.py +++ b/freqtrade/freqai/data_drawer.py @@ -7,17 +7,19 @@ import threading from pathlib import Path from typing import Any, Dict, Tuple -import numpy.typing as npt import numpy as np +import numpy.typing as npt import pandas as pd +from joblib import dump, load from joblib.externals import cloudpickle from pandas import DataFrame -from freqtrade.freqai.data_kitchen import FreqaiDataKitchen -from joblib import dump, load -from freqtrade.strategy.interface import IStrategy -from freqtrade.exceptions import OperationalException -from freqtrade.data.history import load_pair_history + from freqtrade.configuration import TimeRange +from freqtrade.data.history import load_pair_history +from freqtrade.exceptions import OperationalException +from freqtrade.freqai.data_kitchen import FreqaiDataKitchen +from freqtrade.strategy.interface import IStrategy + logger = logging.getLogger(__name__) diff --git a/tests/freqai/conftest.py b/tests/freqai/conftest.py index c039222c0..ff70a40a4 100644 --- a/tests/freqai/conftest.py +++ b/tests/freqai/conftest.py @@ -5,8 +5,8 @@ import pytest from freqtrade.configuration import TimeRange from freqtrade.data.dataprovider import DataProvider -from freqtrade.freqai.data_kitchen import FreqaiDataKitchen from freqtrade.freqai.data_drawer import FreqaiDataDrawer +from freqtrade.freqai.data_kitchen import FreqaiDataKitchen from freqtrade.resolvers import StrategyResolver from freqtrade.resolvers.freqaimodel_resolver import FreqaiModelResolver from tests.conftest import get_patched_exchange