From cb17b36981ba8513c5d0a370864b0cfe312c0c52 Mon Sep 17 00:00:00 2001 From: Yinon Polak Date: Sun, 12 Mar 2023 14:50:08 +0200 Subject: [PATCH] simplify file_type check comparisons --- freqtrade/freqai/data_drawer.py | 3 +-- freqtrade/freqai/freqai_interface.py | 3 +-- 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/freqtrade/freqai/data_drawer.py b/freqtrade/freqai/data_drawer.py index 8d31586fe..fd839ad2f 100644 --- 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: diff --git a/freqtrade/freqai/freqai_interface.py b/freqtrade/freqai/freqai_interface.py index 79bd7d672..3d06745f9 100644 --- a/freqtrade/freqai/freqai_interface.py +++ b/freqtrade/freqai/freqai_interface.py @@ -564,8 +564,7 @@ class IFreqaiModel(ABC): elif self.dd.model_type == 'keras': file_type = ".h5" elif ('stable_baselines' in self.dd.model_type or - 'sb3_contrib' == self.dd.model_type or - 'pytorch' == self.dd.model_type): + self.dd.model_type in ['sb3_contrib', 'pytorch']): file_type = ".zip" path_to_modelfile = Path(dk.data_path / f"{dk.model_filename}_model{file_type}")