Move total_bins out

tanghaibao · Apr 28, 2024 · 3f1c377 · 3f1c377
1 parent 4a4f99b
commit 3f1c377
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/jcvi/assembly/hic.py b/jcvi/assembly/hic.py
@@ -733,6 +733,7 @@ def heatmap(args):
     logger.debug("Resolution set to %d", resolution)
     # Load the matrix
     A = np.load(npyfile)
+    total_bins = header["total_bins"]
 
     # Select specific submatrix
     if contig:
@@ -741,7 +742,6 @@ def heatmap(args):
         contig_end = contig_start + contig_size
         A = A[contig_start:contig_end, contig_start:contig_end]
     else:
-        total_bins = header["total_bins"]
         A = A[:total_bins, :total_bins]
 
     # Convert seqids to positions for each group