Lint fix

nubank · Nov 7, 2023 · d09a4f4 · d09a4f4
1 parent 32dd710
commit d09a4f4
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/src/fklearn/training/classification.py b/src/fklearn/training/classification.py
@@ -16,6 +16,7 @@
 if TYPE_CHECKING:
     from lightgbm import Booster
 
+
 @curry
 @log_learner_time(learner_name='logistic_classification_learner')
 def logistic_classification_learner(df: pd.DataFrame,
@@ -525,9 +526,9 @@ def lgbm_classification_learner(
             Union[Callable[[np.ndarray[Any, Any], Any], Tuple[str, float, bool]],
                   Callable[[np.ndarray[Any, Any], Any], List[Tuple[str, float, bool]]]],
             List[Union[Callable[[np.ndarray[Any, Any], Any],
-            Tuple[str, float, bool]],
-            Callable[[np.ndarray[Any, Any], Any],
-            List[Tuple[str, float, bool]]]]],
+                       Tuple[str, float, bool]],
+                       Callable[[np.ndarray[Any, Any], Any],
+                       List[Tuple[str, float, bool]]]]],
             None
         ]] = None,
         init_model: Optional[Union[str, Path, 'Booster']] = None,

diff --git a/src/fklearn/training/regression.py b/src/fklearn/training/regression.py
@@ -8,7 +8,7 @@
 from sklearn import __version__ as sk_version
 
 from fklearn.common_docstrings import learner_pred_fn_docstring, learner_return_docstring
-from fklearn.types import LearnerReturnType, LearnerLogType
+from fklearn.types import LearnerReturnType
 from fklearn.training.utils import log_learner_time, expand_features_encoded