adelorenzo/stable
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix flake8 found issues

Browse Source
This commit is contained in:
Zohar Kol 2023-03-30 08:25:51 +03:00
parent 4aa9284737
commit 1d8327bb95
3 changed files with 15 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
freqtrade/freqai/data_kitchen.py
View File

@ -112,7 +112,7 @@ class FreqaiDataKitchen:
self.unique_class_list: list = [] self.unique_class_list: list = []
self.backtest_live_models_data: Dict[str, Any] = {} self.backtest_live_models_data: Dict[str, Any] = {}
self.normalizer: Normalization = normalization_factory(config, self.data, self.pkl_data, self.normalizer: Normalization = normalization_factory(config, self.data, self.pkl_data,
self.unique_class_list) self.unique_class_list)
def set_paths( def set_paths(
self, self,

29
freqtrade/freqai/normalization.py
View File

@ -16,20 +16,19 @@ def normalization_factory(
config: Config, config: Config,
meta_data: Dict[str, Any], meta_data: Dict[str, Any],
pickle_meta_data: Dict[str, Any], pickle_meta_data: Dict[str, Any],
unique_class_list: list unique_class_list: list):
): freqai_config: Dict[str, Any] = config["freqai"]
freqai_config: Dict[str, Any] = config["freqai"] norm_config_id = freqai_config["feature_parameters"].get("data_normalization", "legacy")
norm_config_id = freqai_config["feature_parameters"].get("data_normalization", "legacy") if norm_config_id.lower() == "legacy":
if norm_config_id.lower() == "legacy": return LegacyNormalization(config, meta_data, pickle_meta_data, unique_class_list)
return LegacyNormalization(config, meta_data, pickle_meta_data, unique_class_list) elif norm_config_id.lower() == "standard":
elif norm_config_id.lower() == "standard": return StandardNormalization(config, meta_data, pickle_meta_data, unique_class_list)
return StandardNormalization(config, meta_data, pickle_meta_data, unique_class_list) elif norm_config_id.lower() == "minmax":
elif norm_config_id.lower() == "minmax": return MinMaxNormalization(config, meta_data, pickle_meta_data, unique_class_list)
return MinMaxNormalization(config, meta_data, pickle_meta_data, unique_class_list) elif norm_config_id.lower() == "quantile":
elif norm_config_id.lower() == "quantile": return QuantileNormalization(config, meta_data, pickle_meta_data, unique_class_list)
return QuantileNormalization(config, meta_data, pickle_meta_data, unique_class_list) else:
else: raise OperationalException(f"Invalid data normalization identifier '{norm_config_id}'")
raise OperationalException(f"Invalid data normalization identifier '{norm_config_id}'")
class Normalization(ABC): class Normalization(ABC):
@ -268,5 +267,3 @@ class QuantileNormalization(SKLearnNormalization):
unique_class_list: list): unique_class_list: list):
super().__init__(config, meta_data, pickle_meta_data, unique_class_list, super().__init__(config, meta_data, pickle_meta_data, unique_class_list,
QuantileTransformer) QuantileTransformer)

1
tests/freqai/test_normalization.py
View File

@ -84,6 +84,7 @@ def test_assertion_invalid_normalization_id(mocker, freqai_conf):
assert str(e_info).startswith("Invalid data normalization identifier"), \ assert str(e_info).startswith("Invalid data normalization identifier"), \
"unexpected exception string" "unexpected exception string"
@pytest.mark.parametrize( @pytest.mark.parametrize(
"config_id", "config_id",
[ [