fix: Try to fix linting

This commit is contained in:
Robert Caulk
2026-02-05 17:24:17 +01:00
parent 783c365c10
commit 6814bc84aa
2 changed files with 4 additions and 2 deletions

View File

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

View File

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