[ARROW-17933] SparseCOOTensor raises error when created with zero elements

• Type: Bug
• Status: In Progress
• Priority: Major
• Resolution: Unresolved
• Affects Version/s: 9.0.0
• Component/s: Python
• Labels: pull-request-available
• Environment: Ubuntu 22, Python 3.10, Pyarrow 9.0.0
• External issue URL: https://github.com/apache/arrow/issues/33141
• Assignee: Rok Mihevc
• Reporter: Bruce Martin
• Created: 04/Oct/22 15:53
• Updated: 11/Jan/23 11:57

Description

Using pyarrow, SparseCOOTensor generates an error when created with zero elements. SparseCSCMatrix and SparseCSRMatrix work fine.

Example:

In [24]: pyarrow._version_
Out[24]: '9.0.0'
In [25]: pyarrow.SparseCSCMatrix.from_scipy(scipy.sparse.csc_matrix(numpy.zeros((2,4)), dtype=numpy.float32)).non_zero_length
Out[25]: 0
In [26]: pyarrow.SparseCSRMatrix.from_scipy(scipy.sparse.csr_matrix(numpy.zeros((2,4)), dtype=numpy.float32)).non_zero_length
Out[26]: 0
In [27]: pyarrow.SparseCOOTensor.from_scipy(scipy.sparse.coo_matrix(numpy.zeros((2,4)), dtype=numpy.float32)).non_zero_length
---------------------------------------------------------------------------
ArrowInvalid Traceback (most recent call last)
Cell In [27], line 1
----> 1 pyarrow.SparseCOOTensor.from_scipy(scipy.sparse.coo_matrix(numpy.zeros((2,4)), dtype=numpy.float32)).non_zero_length
File ~/venv/lib/python3.10/site-packages/pyarrow/tensor.pxi:400, in pyarrow.lib.SparseCOOTensor.from_scipy()
File ~/venv/lib/python3.10/site-packages/pyarrow/error.pxi:100, in pyarrow.lib.check_status()
ArrowInvalid: SparseCOOIndex indices must be contiguous

Issue Links

• links to: GitHub Pull Request #14378