resolving bug when index is not ordinal number

dongso · Jan 13, 2022 · 0fff3a5 · 0fff3a5
1 parent 8e0a2a1
commit 0fff3a5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pycaret/internal/tabular.py b/pycaret/internal/tabular.py
@@ -10419,7 +10419,7 @@ def check_fairness(estimator, sensitive_features: list, plot_kwargs: dict = {}):
 
     y_pred = estimator.predict(get_config("X_test"))
     y_true = np.array(get_config("y_test"))
-    X_test_before_transform = get_config("data_before_preprocess").iloc[
+    X_test_before_transform = get_config("data_before_preprocess").loc[
         get_config("X_test").index
     ]