fixed #1185, binary colormap (#1186)

librosa · Jun 25, 2020 · 3c57f7c · 3c57f7c
1 parent 6f609e6
commit 3c57f7c
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/librosa/display.py b/librosa/display.py
@@ -340,7 +340,7 @@ def cmap(data, robust=True, cmap_seq='magma', cmap_bool='gray_r', cmap_div='cool
     data = np.atleast_1d(data)
 
     if data.dtype == 'bool':
-        return get_cmap(cmap_bool)
+        return get_cmap(cmap_bool, lut=2)
 
     data = data[np.isfinite(data)]
 
@@ -349,8 +349,7 @@ def cmap(data, robust=True, cmap_seq='magma', cmap_bool='gray_r', cmap_div='cool
     else:
         min_p, max_p = 0, 100
 
-    max_val = np.percentile(data, max_p)
-    min_val = np.percentile(data, min_p)
+    min_val, max_val = np.percentile(data, [min_p, max_p])
 
     if min_val >= 0 or max_val <= 0:
         return get_cmap(cmap_seq)