diff --git a/freqtrade/freqai/RL/Base4ActionRLEnv.py b/freqtrade/freqai/RL/Base4ActionRLEnv.py index 0c719ea92..1a235801c 100644 --- a/freqtrade/freqai/RL/Base4ActionRLEnv.py +++ b/freqtrade/freqai/RL/Base4ActionRLEnv.py @@ -103,7 +103,7 @@ class Base4ActionRLEnv(BaseEnvironment): return observation, step_reward, self._done, info - def is_tradesignal(self, action: int): + def is_tradesignal(self, action: int) -> bool: """ Determine if the signal is a trade signal e.g.: agent wants a Actions.Long_exit while it is in a Positions.short @@ -117,7 +117,7 @@ class Base4ActionRLEnv(BaseEnvironment): (action == Actions.Long_enter.value and self._position == Positions.Long) or (action == Actions.Long_enter.value and self._position == Positions.Short)) - def _is_valid(self, action: int): + def _is_valid(self, action: int) -> bool: """ Determine if the signal is valid. e.g.: agent wants a Actions.Long_exit while it is in a Positions.short diff --git a/freqtrade/freqai/RL/Base5ActionRLEnv.py b/freqtrade/freqai/RL/Base5ActionRLEnv.py index 8012ff1af..61abb8031 100644 --- a/freqtrade/freqai/RL/Base5ActionRLEnv.py +++ b/freqtrade/freqai/RL/Base5ActionRLEnv.py @@ -141,7 +141,7 @@ class Base5ActionRLEnv(BaseEnvironment): else: return self._current_tick - self._last_trade_tick - def is_tradesignal(self, action: int): + def is_tradesignal(self, action: int) -> bool: """ Determine if the signal is a trade signal e.g.: agent wants a Actions.Long_exit while it is in a Positions.short diff --git a/freqtrade/freqai/RL/BaseEnvironment.py b/freqtrade/freqai/RL/BaseEnvironment.py index 7aa571697..3332e5a18 100644 --- a/freqtrade/freqai/RL/BaseEnvironment.py +++ b/freqtrade/freqai/RL/BaseEnvironment.py @@ -200,13 +200,13 @@ class BaseEnvironment(gym.Env): return 0. @abstractmethod - def is_tradesignal(self, action: int): + def is_tradesignal(self, action: int) -> bool: """ Determine if the signal is a trade signal. This is unique to the actions in the environment, and therefore must be inherited. """ - return + return True def _is_valid(self, action: int) -> bool: """ diff --git a/freqtrade/freqai/RL/BaseReinforcementLearningModel.py b/freqtrade/freqai/RL/BaseReinforcementLearningModel.py index bddac23b3..af9874d90 100644 --- a/freqtrade/freqai/RL/BaseReinforcementLearningModel.py +++ b/freqtrade/freqai/RL/BaseReinforcementLearningModel.py @@ -38,7 +38,7 @@ class BaseReinforcementLearningModel(IFreqaiModel): User created Reinforcement Learning Model prediction class """ - def __init__(self, **kwargs): + def __init__(self, **kwargs) -> None: super().__init__(config=kwargs['config']) self.max_threads = min(self.freqai_info['rl_config'].get( 'cpu_count', 1), max(int(self.max_system_threads / 2), 1))