Skip to content

GitLab

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
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
torch_sparse/storage.py
View file @ c0805819
...@@ -48,7 +48,7 @@ class SparseStorage(object): ...@@ -48,7 +48,7 @@ class SparseStorage(object):
assert col.dim() == 1 assert col.dim() == 1
col = col.contiguous() 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: if rowptr is not None:
M = rowptr.numel() - 1 M = rowptr.numel() - 1
elif row is not None and row.numel() > 0: 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