Black is fighting against mypy

AnotherSamWilson · Jul 26, 2024 · 6a0906d · 6a0906d
1 parent 8967ff8
commit 6a0906d
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/miceforest/imputation_kernel.py b/miceforest/imputation_kernel.py
@@ -798,9 +798,8 @@ def _get_candidate_preds_mice(
             candidate_preds = lgbmodel.predict(
                 candidate_features,
                 pred_contrib=True,
-            ).astype(
-                _PRE_LINK_DATATYPE
-            )  # type: ignore
+            )
+            candidate_preds = candidate_preds.astype(_PRE_LINK_DATATYPE)  # type: ignore
         else:
             candidate_preds = lgbmodel._Booster__inner_predict(0)  # type: ignore
             if logistic and not (shap or fast):