typo

c2bd0b10 · rusty1s · 62891fa0 · c2bd0b10
Commit c2bd0b10 authored Dec 20, 2019 by rusty1s
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

torch_sparse/index_select.py torch_sparse/index_select.py +2 -2

No files found.
--- a/torch_sparse/index_select.py
+++ b/torch_sparse/index_select.py
@@ -57,8 +57,8 @@ def index_select(src, dim, idx):
        sparse_size = torch.Size([src.sparse_size(0), colcount.size(0)])

        storage = src.storage.__class__(index, value, sparse_size,
-                                        colcount=colcount, csc2csr=csc2csr,
-                                        is_sorted=True)
+                                        colcount=colcount, colptr=colptr,
+                                        csc2csr=csc2csr, is_sorted=True)

    else:
        storage = src.storage.apply_value(