lychiyu/freqtrade_origin
1
0
Fork 0
mirror of https://github.com/freqtrade/freqtrade.git synced 2024-09-20 09:31:12 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

try even more deactivation

Browse Source
This commit is contained in:
robcaulk 2023-05-14 15:49:24 +00:00
parent 1ed084557a
commit 505f36a95f
1 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
freqtrade/freqai/utils.py
View File

@ -1,7 +1,7 @@
import logging
from datetime import datetime, timezone
from pathlib import Path
from typing import Any, Dict, Union
from typing import Any, Dict
import numpy as np
import pandas as pd
@ -16,8 +16,6 @@ from freqtrade.exchange import timeframe_to_seconds
from freqtrade.exchange.exchange import market_is_active
from freqtrade.freqai.data_drawer import FreqaiDataDrawer
from freqtrade.freqai.data_kitchen import FreqaiDataKitchen
from freqtrade.freqai.tensorboard import TBLogger
from freqtrade.freqai.tensorboard.base_tensorboard import BaseTensorboardLogger
from freqtrade.plugins.pairlist.pairlist_helpers import dynamic_expand_pairlist
@ -188,12 +186,13 @@ def get_timerange_backtest_live_models(config: Config) -> str:
return timerange.timerange_str
def get_tb_logger(model_type: str, path: Path, activate: bool) -> Union[TBLogger,
BaseTensorboardLogger]:
tb_logger: Union[TBLogger, BaseTensorboardLogger]
def get_tb_logger(model_type: str, path: Path, activate: bool) -> Any:
if model_type == "pytorch" and activate:
from freqtrade.freqai.tensorboard import TBLogger
tb_logger = TBLogger(path, activate)
else:
tb_logger = BaseTensorboardLogger(path, activate)
from freqtrade.freqai.tensorboard.base_tensorboard import BaseTensorboardLogger
tb_logger = BaseTensorboardLogger(path, activate) # type: ignore
return tb_logger