Merge pull request #323 from robomics/bugfix

Improve handling of uint attributes

cooler/balance.py

@@ -340,7 +340,7 @@ def balance_cooler(
 
     """
     # Divide the number of elements into non-overlapping chunks
-    nnz = clr.info["nnz"]
+    nnz = int(clr.info["nnz"])
     if chunksize is None:
         chunksize = nnz
         spans = [(0, nnz)]
@@ -356,7 +356,7 @@ def balance_cooler(
         base_filters.append(partial(_zero_diags, ignore_diags))
 
     # Initialize the bias weights
-    n_bins = clr.info["nbins"]
+    n_bins = int(clr.info["nbins"])
     if x0 is not None:
         bias = x0
         bias[np.isnan(bias)] = 0

cooler/cli/show.py

@@ -100,7 +100,7 @@ def update_heatmap(event):
         im.set_extent(extent)
         ax.figure.canvas.draw_idle()
 
-    binsize = c.info["bin-size"]
+    binsize = int(c.info["bin-size"])
     row_chrom_len = c.chromsizes[row_chrom]
     col_chrom_len = c.chromsizes[col_chrom]
     plotstate = {"placeholders": [], "prev_extent": get_extent(plt.gca())}

cooler/parallel.py

@@ -268,5 +268,5 @@ def __call__(self, span):
 
 def split(clr, map=map, chunksize=10_000_000, spans=None, **kwargs):
     if spans is None:
-        spans = partition(0, clr.info["nnz"], chunksize)
+        spans = partition(0, int(clr.info["nnz"]), chunksize)
     return MultiplexDataPipe(chunkgetter(clr, **kwargs), spans, map)