From c6f3a3bbca4472918c8a60a2784075936621483a Mon Sep 17 00:00:00 2001 From: robcaulk Date: Wed, 26 Apr 2023 14:11:26 +0200 Subject: [PATCH] avoid typing issues in the tensorboard callback --- freqtrade/freqai/RL/TensorboardCallback.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/freqtrade/freqai/RL/TensorboardCallback.py b/freqtrade/freqai/RL/TensorboardCallback.py index 282e60b0d..3924f9d2c 100644 --- a/freqtrade/freqai/RL/TensorboardCallback.py +++ b/freqtrade/freqai/RL/TensorboardCallback.py @@ -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: