Fix broken classes attribute

Renumics · Dec 4, 2023 · 217f469 · 217f469
1 parent 3d6fb29
commit 217f469
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()