simplify file_type check comparisons

2023-03-12 14:50:08 +02:00
parent f9fdf1c31b
commit cb17b36981
2 changed files with 2 additions and 4 deletions
--- a/freqtrade/freqai/data_drawer.py
+++ b/freqtrade/freqai/data_drawer.py
@@ -447,8 +447,7 @@ class FreqaiDataDrawer:
        elif self.model_type == 'keras':
            model.save(save_path / f"{dk.model_filename}_model.h5")
        elif ('stable_baselines' in self.model_type or
-              'sb3_contrib' == self.model_type or
-              'pytorch' == self.model_type):
+              self.model_type in ['sb3_contrib', 'pytorch']):
            model.save(save_path / f"{dk.model_filename}_model.zip")

        if dk.svm_model is not None: