From 0357d373a91b3af108184fb5a971c8357689e05a Mon Sep 17 00:00:00 2001
From: initrv <37817561+initrv@users.noreply.github.com>
Date: Fri, 1 Sep 2023 11:31:09 +0300
Subject: [PATCH] add QRDQN to SB3_CONTRIB_MODELS

---
 freqtrade/freqai/RL/BaseReinforcementLearningModel.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/freqtrade/freqai/RL/BaseReinforcementLearningModel.py b/freqtrade/freqai/RL/BaseReinforcementLearningModel.py
index 06ce772a8..5b98efb49 100644
--- a/freqtrade/freqai/RL/BaseReinforcementLearningModel.py
+++ b/freqtrade/freqai/RL/BaseReinforcementLearningModel.py
@@ -33,7 +33,7 @@ logger = logging.getLogger(__name__)
 torch.multiprocessing.set_sharing_strategy('file_system')
 
 SB3_MODELS = ['PPO', 'A2C', 'DQN']
-SB3_CONTRIB_MODELS = ['TRPO', 'ARS', 'RecurrentPPO', 'MaskablePPO']
+SB3_CONTRIB_MODELS = ['TRPO', 'ARS', 'RecurrentPPO', 'MaskablePPO', 'QRDQN']
 
 
 class BaseReinforcementLearningModel(IFreqaiModel):