Lint fix

nubank · Jul 10, 2024 · 4ad7e05 · 4ad7e05
1 parent eafe0cd
commit 4ad7e05
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 18 deletions.
diff --git a/install_all_deps.sh b/install_all_deps.sh
diff --git a/src/fklearn/validation/evaluators.py b/src/fklearn/validation/evaluators.py
@@ -1065,7 +1065,9 @@ def logistic_coefficient_evaluator(test_data: pd.DataFrame,
     if eval_name is None:
         eval_name = "logistic_coefficient_evaluator__" + target_column
 
-    score = LogisticRegression(penalty=None, multi_class="ovr").fit(test_data[[prediction_column]],
-                                                                      test_data[target_column]).coef_[0][0]
+    score = LogisticRegression(penalty=None, multi_class="ovr").fit(
+        test_data[[prediction_column]],
+        test_data[target_column]
+    ).coef_[0][0]
 
     return {eval_name: score}