ap fix

NannyML · Apr 24, 2024 · 22eb707 · 22eb707
1 parent 1f1dbe8
commit 22eb707
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nannyml/performance_estimation/confidence_based/metrics.py b/nannyml/performance_estimation/confidence_based/metrics.py
@@ -520,7 +520,7 @@ def _fit(self, reference_data: pd.DataFrame):
         y_pred_proba = data[self.y_pred_proba]
 
         # if empty then positive class won't be part of y_true series
-        if empty:
+        if 1 not in y_true.unique():
             self._logger.debug(f"Not enough data to compute fit {self.display_name}.")
             warnings.warn(f"Not enough data to compute fit {self.display_name}.")
             self._sampling_error_components = np.NaN, 0