Merge pull request #65 from omerferhatt/patch-1

Fix broken `classes` attribute
Renumics · Dec 4, 2023 · ba15c30 · ba15c30
2 parents 3d6fb29 + 217f469
commit ba15c30
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sliceguard/sliceguard.py b/sliceguard/sliceguard.py
@@ -428,7 +428,7 @@ def report(
             df["sg_y_pred"] = self._generated_y_pred[selected_dataframe_rows]
 
         if hasattr(self, "_generated_y_probs") and hasattr(self, "classes"):
-            for class_idx, label in enumerate(self._classes):
+            for class_idx, label in enumerate(self.classes):
                 df[f"sg_p_{label}"] = self._generated_y_probs[:, class_idx].tolist()
 
         spotlight_issue_list = np.array(data_issues)[data_issue_order].tolist()