black formatting

AnotherSamWilson · Jul 26, 2024 · 8967ff8 · 8967ff8
1 parent d8374b2
commit 8967ff8
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/miceforest/imputation_kernel.py b/miceforest/imputation_kernel.py
@@ -560,9 +560,9 @@ def _get_oof_performance(
                 log_evaluation(period=0),
             ],
         )
-        best_iteration = lgbcv["cvbooster"].best_iteration # type: ignore
+        best_iteration = lgbcv["cvbooster"].best_iteration  # type: ignore
         loss_metric_key = list(lgbcv)[0]
-        loss: float = np.min(lgbcv[loss_metric_key]) # type: ignore
+        loss: float = np.min(lgbcv[loss_metric_key])  # type: ignore
 
         return loss, best_iteration
 
@@ -798,9 +798,11 @@ def _get_candidate_preds_mice(
             candidate_preds = lgbmodel.predict(
                 candidate_features,
                 pred_contrib=True,
-            ).astype(_PRE_LINK_DATATYPE) # type: ignore
+            ).astype(
+                _PRE_LINK_DATATYPE
+            )  # type: ignore
         else:
-            candidate_preds = lgbmodel._Booster__inner_predict(0) # type: ignore
+            candidate_preds = lgbmodel._Booster__inner_predict(0)  # type: ignore
             if logistic and not (shap or fast):
                 candidate_preds = logodds(candidate_preds).astype(_PRE_LINK_DATATYPE)