lychiyu/freqtrade_origin
1
0
Fork 0
mirror of https://github.com/freqtrade/freqtrade.git synced 2024-11-10 10:21:59 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

avoid typing issues in the tensorboard callback

Browse Source
This commit is contained in:
robcaulk 2023-04-26 14:11:26 +02:00
parent e86980befa
commit c6f3a3bbca
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
freqtrade/freqai/RL/TensorboardCallback.py
View File

@ -3,9 +3,8 @@ from typing import Any, Dict, Type, Union
from stable_baselines3.common.callbacks import BaseCallback
from stable_baselines3.common.logger import HParam
from stable_baselines3.common.vec_env import DummyVecEnv, SubprocVecEnv, VecMonitor
from freqtrade.freqai.RL.BaseEnvironment import BaseActions, BaseEnvironment
from freqtrade.freqai.RL.BaseEnvironment import BaseActions
class TensorboardCallback(BaseCallback):
@ -17,8 +16,6 @@ class TensorboardCallback(BaseCallback):
super().__init__(verbose)
self.model: Any = None
self.logger = None # type: Any
self.training_env: Union[BaseEnvironment, SubprocVecEnv,
VecMonitor, DummyVecEnv] = DummyVecEnv()
self.actions: Type[Enum] = actions
def _on_training_start(self) -> None: