adelorenzo/stable
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

isort

Browse Source
This commit is contained in:
robcaulk 2022-09-23 19:30:56 +02:00
parent 77c360b264
commit 647200e8a7
3 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
freqtrade/freqai/RL/BaseReinforcementLearningModel.py
View File

@ -14,6 +14,7 @@ from pandas import DataFrame
from stable_baselines3.common.callbacks import EvalCallback
from stable_baselines3.common.monitor import Monitor
from stable_baselines3.common.utils import set_random_seed
from stable_baselines3.common.vec_env import SubprocVecEnv
from freqtrade.exceptions import OperationalException
from freqtrade.freqai.data_kitchen import FreqaiDataKitchen
@ -21,7 +22,6 @@ from freqtrade.freqai.freqai_interface import IFreqaiModel
from freqtrade.freqai.RL.Base5ActionRLEnv import Actions, Base5ActionRLEnv
from freqtrade.freqai.RL.BaseEnvironment import Positions
from freqtrade.persistence import Trade
from stable_baselines3.common.vec_env import SubprocVecEnv
logger = logging.getLogger(__name__)

3
freqtrade/freqai/prediction_models/ReinforcementLearner_multiproc.py
View File

@ -4,9 +4,10 @@ from typing import Any, Dict # , Tuple
# import numpy.typing as npt
import torch as th
from pandas import DataFrame
from stable_baselines3.common.callbacks import EvalCallback
from stable_baselines3.common.vec_env import SubprocVecEnv
from pandas import DataFrame
from freqtrade.freqai.data_kitchen import FreqaiDataKitchen
from freqtrade.freqai.RL.BaseReinforcementLearningModel import (BaseReinforcementLearningModel,
make_env)

7
tests/freqai/test_freqai_interface.py
View File

@ -4,15 +4,16 @@ from pathlib import Path
from unittest.mock import MagicMock
import pytest
from freqtrade.enums import RunMode
from freqtrade.configuration import TimeRange
from freqtrade.data.dataprovider import DataProvider
from freqtrade.enums import RunMode
from freqtrade.freqai.data_kitchen import FreqaiDataKitchen
from freqtrade.freqai.utils import download_all_data_for_training, get_required_data_timerange
from freqtrade.persistence import Trade
from freqtrade.plugins.pairlistmanager import PairListManager
from tests.conftest import get_patched_exchange, log_has_re
from tests.freqai.conftest import get_patched_freqai_strategy
from freqtrade.persistence import Trade
from freqtrade.freqai.utils import download_all_data_for_training, get_required_data_timerange
def is_arm() -> bool: