Fix bug in pymaf.MafFrame.plot_waterfall (#67):

* :issue:`67`: Fix bug in :meth:`pymaf.MafFrame.plot_waterfall` method where ``count=1`` was causing color mismatch.
sbslee · Mar 23, 2023 · 5ae6623 · 5ae6623
1 parent fddeb64
commit 5ae6623
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/CHANGELOG.rst b/CHANGELOG.rst
@@ -4,6 +4,8 @@ Changelog
 0.37.0 (in development)
 -----------------------
 
+* :issue:`67`: Fix bug in :meth:`pymaf.MafFrame.plot_waterfall` method where ``count=1`` was causing color mismatch.
+
 0.36.0 (2022-08-12)
 -------------------
 

diff --git a/fuc/api/pymaf.py b/fuc/api/pymaf.py
@@ -3263,7 +3263,7 @@ def plot_waterfall(
  colors = list(reversed(NONSYN_COLORS + ['k', 'lightgray']))
 
  sns.heatmap(df, cmap=colors, ax=ax, xticklabels=False,
- cbar=False, **kwargs)
+ cbar=False, vmin=0, vmax=len(colors), **kwargs)
 
  ax.set_xlabel('Samples')
  ax.set_ylabel('')