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

remove typing from callback init

Browse Source
This commit is contained in:
robcaulk 2023-04-26 13:42:10 +02:00
parent e29ce218eb
commit e86980befa
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -3,7 +3,7 @@ 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 SubprocVecEnv, VecMonitor
from stable_baselines3.common.vec_env import DummyVecEnv, SubprocVecEnv, VecMonitor
from freqtrade.freqai.RL.BaseEnvironment import BaseActions, BaseEnvironment
@ -17,7 +17,8 @@ class TensorboardCallback(BaseCallback):
super().__init__(verbose)
self.model: Any = None
self.logger = None # type: Any
self.training_env: Union[BaseEnvironment, SubprocVecEnv, VecMonitor] = None
self.training_env: Union[BaseEnvironment, SubprocVecEnv,
VecMonitor, DummyVecEnv] = DummyVecEnv()
self.actions: Type[Enum] = actions
def _on_training_start(self) -> None: