Merge pull request #8690 from freqtrade/remove-tb-warning

Update base_tensorboard.py
This commit is contained in:
Matthias 2023-05-25 18:21:05 +02:00 committed by GitHub
commit 61ee77e07e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -10,8 +10,7 @@ logger = logging.getLogger(__name__)
class BaseTensorboardLogger: class BaseTensorboardLogger:
def __init__(self, logdir: Path, activate: bool = True): def __init__(self, logdir: Path, activate: bool = True):
logger.warning("Tensorboard is not installed, no logs will be written." pass
"Ensure torch is installed, or use the torch/RL docker images")
def log_scalar(self, tag: str, scalar_value: Any, step: int): def log_scalar(self, tag: str, scalar_value: Any, step: int):
return return
@ -23,8 +22,7 @@ class BaseTensorboardLogger:
class BaseTensorBoardCallback(TrainingCallback): class BaseTensorBoardCallback(TrainingCallback):
def __init__(self, logdir: Path, activate: bool = True): def __init__(self, logdir: Path, activate: bool = True):
logger.warning("Tensorboard is not installed, no logs will be written." pass
"Ensure torch is installed, or use the torch/RL docker images")
def after_iteration( def after_iteration(
self, model, epoch: int, evals_log: TrainingCallback.EvalsLog self, model, epoch: int, evals_log: TrainingCallback.EvalsLog