Merge branch 'main' of github.com:Renumics/sliceguard into main

Renumics · Dec 4, 2023 · cdef579 · cdef579
2 parents 9ba2ae4 + ba15c30
commit cdef579
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()