cheat flake8 for now until we can refactor save into the model class

2023-03-14 21:13:30 +01:00
parent 244662b1a4
commit fab505be1b
3 changed files with 4 additions and 4 deletions
--- a/freqtrade/freqai/prediction_models/PyTorchClassifierMultiTarget.py
+++ b/freqtrade/freqai/prediction_models/PyTorchClassifierMultiTarget.py
@@ -41,7 +41,7 @@ class PyTorchClassifierMultiTarget(BasePyTorchModel):
        self.max_n_eval_batches: Optional[int] = model_training_params.get(
            "max_n_eval_batches", None
        )
-        self.model_kwargs: Dict = model_training_params.get("model_kwargs", {})
+        self.model_kwargs: Dict[str, any] = model_training_params.get("model_kwargs", {})
        self.class_name_to_index = None
        self.index_to_class_name = None