diff --git a/freqtrade/freqai/prediction_models/LightGBMClassifier.py b/freqtrade/freqai/prediction_models/LightGBMClassifier.py index 75bbe2d1a..0b78c4129 100644 --- a/freqtrade/freqai/prediction_models/LightGBMClassifier.py +++ b/freqtrade/freqai/prediction_models/LightGBMClassifier.py @@ -1,4 +1,5 @@ import logging +from collections.abc import Callable from typing import Any from lightgbm import LGBMClassifier @@ -48,7 +49,7 @@ class LightGBMClassifier(BaseClassifierModel): model = LGBMClassifier(**self.model_training_parameters) activate_tensorboard = self.freqai_info.get("activate_tensorboard", True) - callbacks = [] + callbacks: list[Callable[..., Any]] = [] if LightGBMCallback is not None: callbacks = [LightGBMCallback(dk.data_path, activate_tensorboard)] model.fit( diff --git a/freqtrade/freqai/prediction_models/LightGBMRegressor.py b/freqtrade/freqai/prediction_models/LightGBMRegressor.py index af89d4825..abd838eee 100644 --- a/freqtrade/freqai/prediction_models/LightGBMRegressor.py +++ b/freqtrade/freqai/prediction_models/LightGBMRegressor.py @@ -1,4 +1,5 @@ import logging +from collections.abc import Callable from typing import Any from lightgbm import LGBMRegressor @@ -44,7 +45,7 @@ class LightGBMRegressor(BaseRegressionModel): model = LGBMRegressor(**self.model_training_parameters) activate_tensorboard = self.freqai_info.get("activate_tensorboard", True) - callbacks = [] + callbacks: list[Callable[..., Any]] = [] if LightGBMCallback is not None: callbacks = [LightGBMCallback(dk.data_path, activate_tensorboard)]