mohammadian7/freqtrade
1
0
Fork 0
Code Issues Pull Requests Actions 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
Download Patch File Download Diff File Expand all files Collapse all files
freqtrade/freqai/RL/TensorboardCallback.py
+1 -4
View File
@@ -3,9 +3,8 @@ 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 DummyVecEnv, SubprocVecEnv, VecMonitor
from freqtrade.freqai.RL.BaseEnvironment import BaseActions, BaseEnvironment from freqtrade.freqai.RL.BaseEnvironment import BaseActions
class TensorboardCallback(BaseCallback): class TensorboardCallback(BaseCallback):
@@ -17,8 +16,6 @@ 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, DummyVecEnv] = DummyVecEnv()
self.actions: Type[Enum] = actions self.actions: Type[Enum] = actions
def _on_training_start(self) -> None: def _on_training_start(self) -> None: