fix flake8 found issues

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

View File

@ -16,8 +16,7 @@ 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":
@ -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)

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",
[ [