typo

rusty1s · rusty1s · commit 3dbd22823684 · 2021-08-15T12:33:01.000+02:00
diff --git a/torch_sparse/tensor.py b/torch_sparse/tensor.py
@@ -435,6 +435,7 @@ def to_torch_sparse_coo_tensor(
 
 def share_memory_(self: SparseTensor) -> SparseTensor:
     self.storage.share_memory_()
+    return self
 
 
 def is_shared(self: SparseTensor) -> bool:
