nnz may not fit in uint8.

pydata · Sep 27, 2024 · e3d2ec1 · e3d2ec1
1 parent 457dd56
commit e3d2ec1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sparse/numba_backend/_coo/core.py b/sparse/numba_backend/_coo/core.py
@@ -1539,15 +1539,15 @@ def __binsparse__(self) -> tuple[dict, list[np.ndarray]]:
                 "shape": list(self.shape),
                 "number_of_stored_values": self.nnz,
                 "data_types": {
-                    "pointers_to_1": "uint8",
+                    "pointers_to_1": "uint64",
                     "indices_1": str(self.coords.dtype),
                     "values": data_dt,
                 },
             },
             "original_source": f"`sparse`, version {__version__}",
         }
 
-        return descriptor, [np.array([0, self.nnz], dtype=np.uint8), self.coords, self.data]
+        return descriptor, [np.array([0, self.nnz], dtype=np.uint64), self.coords, self.data]
 
 
 def as_coo(x, shape=None, fill_value=None, idx_dtype=None):