Unverified Commit c0805819 authored by Feng Shi's avatar Feng Shi Committed by GitHub
Browse files

Update storage.py

BUG: when sparse_sizes is passed with (None, None), this causes an error
parent 709f6837
......@@ -48,7 +48,7 @@ class SparseStorage(object):
assert col.dim() == 1
col = col.contiguous()
if sparse_sizes is None:
if sparse_sizes is None or sparse_sizes[0] is None or sparse_sizes[1] is None:
if rowptr is not None:
M = rowptr.numel() - 1
elif row is not None and row.numel() > 0:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment