Skip to content

GitLab

Commit a1ae9033 authored by rusty1s's avatar rusty1s
Browse files

typo

parent 4c8a4155
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
torch_sparse/masked_select.py
View file @ a1ae9033
...@@ -91,6 +91,6 @@ def masked_select_nnz(src: SparseTensor, mask: torch.Tensor, ...@@ -91,6 +91,6 @@ def masked_select_nnz(src: SparseTensor, mask: torch.Tensor,
SparseTensor.masked_select = lambda self, dim, mask: masked_select( SparseTensor.masked_select = lambda self, dim, mask: masked_select(
self, dim, mask) self, dim, mask)
tmp = lambda src, mask, layout=None: masked_select_nnz( # noqa tmp = lambda self, mask, layout=None: masked_select_nnz( # noqa
src, mask, layout) self, mask, layout)
SparseTensor.masked_select_nnz = tmp SparseTensor.masked_select_nnz = tmp
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