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.backtest_live_models_data: Dict[str, Any] = {}
self.normalizer: Normalization = normalization_factory(config, self.data, self.pkl_data,
self.unique_class_list)
self.unique_class_list)
def set_paths(
self,

29
freqtrade/freqai/normalization.py
View File

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