fix: Try to fix linting
This commit is contained in:
@@ -1,4 +1,5 @@
|
|||||||
import logging
|
import logging
|
||||||
|
from collections.abc import Callable
|
||||||
from typing import Any
|
from typing import Any
|
||||||
|
|
||||||
from lightgbm import LGBMClassifier
|
from lightgbm import LGBMClassifier
|
||||||
@@ -48,7 +49,7 @@ class LightGBMClassifier(BaseClassifierModel):
|
|||||||
|
|
||||||
model = LGBMClassifier(**self.model_training_parameters)
|
model = LGBMClassifier(**self.model_training_parameters)
|
||||||
activate_tensorboard = self.freqai_info.get("activate_tensorboard", True)
|
activate_tensorboard = self.freqai_info.get("activate_tensorboard", True)
|
||||||
callbacks = []
|
callbacks: list[Callable[..., Any]] = []
|
||||||
if LightGBMCallback is not None:
|
if LightGBMCallback is not None:
|
||||||
callbacks = [LightGBMCallback(dk.data_path, activate_tensorboard)]
|
callbacks = [LightGBMCallback(dk.data_path, activate_tensorboard)]
|
||||||
model.fit(
|
model.fit(
|
||||||
|
|||||||
@@ -1,4 +1,5 @@
|
|||||||
import logging
|
import logging
|
||||||
|
from collections.abc import Callable
|
||||||
from typing import Any
|
from typing import Any
|
||||||
|
|
||||||
from lightgbm import LGBMRegressor
|
from lightgbm import LGBMRegressor
|
||||||
@@ -44,7 +45,7 @@ class LightGBMRegressor(BaseRegressionModel):
|
|||||||
model = LGBMRegressor(**self.model_training_parameters)
|
model = LGBMRegressor(**self.model_training_parameters)
|
||||||
|
|
||||||
activate_tensorboard = self.freqai_info.get("activate_tensorboard", True)
|
activate_tensorboard = self.freqai_info.get("activate_tensorboard", True)
|
||||||
callbacks = []
|
callbacks: list[Callable[..., Any]] = []
|
||||||
if LightGBMCallback is not None:
|
if LightGBMCallback is not None:
|
||||||
callbacks = [LightGBMCallback(dk.data_path, activate_tensorboard)]
|
callbacks = [LightGBMCallback(dk.data_path, activate_tensorboard)]
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user