Add can_short param to base env

2022-12-16 22:16:19 +03:00
parent 439914caef
commit dde363343c
3 changed files with 6 additions and 2 deletions
--- a/freqtrade/freqai/RL/BaseReinforcementLearningModel.py
+++ b/freqtrade/freqai/RL/BaseReinforcementLearningModel.py
@@ -165,7 +165,8 @@ class BaseReinforcementLearningModel(IFreqaiModel):
        env_info = {"window_size": self.CONV_WIDTH,
                    "reward_kwargs": self.reward_params,
                    "config": self.config,
-                    "live": self.live}
+                    "live": self.live,
+                    "can_short": self.can_short}
        if self.data_provider:
            env_info["fee"] = self.data_provider._exchange \
                .get_fee(symbol=self.data_provider.current_whitelist()[0])  # type: ignore