fix problem when plotting per chromosome if the chromosome (usually a…

…n scaffold) is too small
deeptools · Jan 11, 2018 · 4e9ae6c · 4e9ae6c
1 parent 831df58
commit 4e9ae6c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/hicexplorer/hicPlotMatrix.py b/hicexplorer/hicPlotMatrix.py
@@ -167,6 +167,9 @@ def change_chrom_names(chrom):
 def plotHeatmap(ma, chrBinBoundaries, fig, position, args, cmap, xlabel=None,
                 ylabel=None, start_pos=None, start_pos2=None, pNorm=None, pAxis=None, pPca=None):
     log.debug("plotting heatmap")
+    if ma.shape[0] < 5:
+        log.info("Matrix for {} too small to plot. Matrix size: {}".format(chrBinBoundaries.keys()[0], ma.shape))
+        return
     if pAxis is not None:
         axHeat2 = pAxis
     else: