CTYPTO/freqtrade
1
0
Fork 0
mirror of https://github.com/freqtrade/freqtrade.git synced 2025-11-29 08:33:07 +00:00
Code Issues 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
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.callbacks import BaseCallback
from stable_baselines3.common.logger import HParam 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 from freqtrade.freqai.RL.BaseEnvironment import BaseActions, BaseEnvironment
@@ -17,7 +17,8 @@ class TensorboardCallback(BaseCallback):
super().__init__(verbose) super().__init__(verbose)
self.model: Any = None self.model: Any = None
self.logger = None # type: Any 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 self.actions: Type[Enum] = actions
def _on_training_start(self) -> None: def _on_training_start(self) -> None: