Update convert.py

pydata · Aug 9, 2019 · 02a8589 · 02a8589
1 parent 5edf2be
commit 02a8589
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/sparse/compressed/convert.py b/sparse/compressed/convert.py
@@ -13,7 +13,8 @@ def convert_to_flat(inds, shape, axisptr):
     shape_bins = transform_shape(shape[axisptr:])
     increments = List()
     for i in range(len(uncompressed_inds)):
-        increments.append(uncompressed_inds[i] * shape_bins[i])
+        increments.append(
+            (uncompressed_inds[i] * shape_bins[i]).astype(np.int32))
     operations = np.prod([ind.shape[0] for ind in increments[:-1]])
     return compute_flat(increments, cols, operations)