black

cleanlab · jwmueller · Jun 23, 2023 · Apr 20, 2023 · Apr 20, 2023 · Apr 21, 2023
commit cde47e64b1bd7b29786cc1be6e7b77f657ebdc5a
diff --git a/cleanlab/segmentation/rank.py b/cleanlab/segmentation/rank.py
@@ -150,9 +150,7 @@ def get_label_quality_scores(
 
 
 def issues_from_scores(
-    image_scores: np.ndarray, 
-    pixel_scores: Optional[np.ndarray] = None, 
-    threshold: float = 0.1
+    image_scores: np.ndarray, pixel_scores: Optional[np.ndarray] = None, threshold: float = 0.1
 ) -> Union[list, np.ndarray]:
     """
     Converts scores output by :py:func:`segmentation.rank.get_label_quality_scores <cleanlab.segmentation.rank.get_label_quality_scores>`

diff --git a/cleanlab/segmentation/summary.py b/cleanlab/segmentation/summary.py
@@ -108,7 +108,6 @@ def display_issues(
             cmap = _generate_colormap(num_classes)
         else:
             cmap = None
-
 
     # Show a legend
     if class_names is not None and cmap is not None:
@@ -146,8 +145,8 @@ def display_issues(
             ax = axes
         else:
             ax = axes[plot_index]
-        
-        mask = np.full((h, w), True) 
+
+        mask = np.full((h, w), True)
         if labels is not None and len(exclude) != 0:
             mask = ~np.isin(labels[i], exclude)
         ax.imshow(issues[i] & mask, cmap=error_cmap, vmin=0, vmax=1)