fix: Try to fix linting

feat/tb-lightgbm
Robert Caulk 2 weeks ago
parent 783c365c10
commit 6814bc84aa

@ -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(

@ -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)]

Loading…
Cancel
Save