Merge pull request #7834 from freqtrade/fix-data-drawer-model-load

Fix sb3_contrib loading issue
2022-12-02 06:29:23 +01:00
parent f2624112b0 396e666e9b
commit 4dc591a170
2 changed files with 4 additions and 4 deletions
--- a/freqtrade/freqai/RL/BaseReinforcementLearningModel.py
+++ b/freqtrade/freqai/RL/BaseReinforcementLearningModel.py
@@ -64,7 +64,7 @@ class BaseReinforcementLearningModel(IFreqaiModel):
        self.policy_type = self.freqai_info['rl_config']['policy_type']
        self.unset_outlier_removal()
        self.net_arch = self.rl_config.get('net_arch', [128, 128])
-        self.dd.model_type = "stable_baselines"
+        self.dd.model_type = import_str

    def unset_outlier_removal(self):
        """