freqtrade_origin/freqtrade/freqai/tensorboard/__init__.py
2024-05-13 07:10:25 +02:00

17 lines
539 B
Python

# ensure users can still use a non-torch freqai version
try:
from freqtrade.freqai.tensorboard.tensorboard import TensorBoardCallback, TensorboardLogger
TBLogger = TensorboardLogger
TBCallback = TensorBoardCallback
except ModuleNotFoundError:
from freqtrade.freqai.tensorboard.base_tensorboard import (
BaseTensorBoardCallback,
BaseTensorboardLogger,
)
TBLogger = BaseTensorboardLogger # type: ignore
TBCallback = BaseTensorBoardCallback # type: ignore
__all__ = ("TBLogger", "TBCallback")