From 328aaf2f92cdc37c8daa2d38c53c995ffbd743c8 Mon Sep 17 00:00:00 2001 From: Maurits Diephuis Date: Tue, 11 Feb 2020 17:34:43 +0100 Subject: [PATCH] fixed a bracket placement --- torch_sparse/coalesce.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torch_sparse/coalesce.py b/torch_sparse/coalesce.py index 1e52e730..985ac74e 100644 --- a/torch_sparse/coalesce.py +++ b/torch_sparse/coalesce.py @@ -20,6 +20,6 @@ def coalesce(index, value, m, n, op="add"): """ storage = SparseStorage(row=index[0], col=index[1], value=value, - sparse_sizes=torch.Size([m, n], is_sorted=False)) + sparse_sizes=torch.Size([m, n]), is_sorted=False) storage = storage.coalesce(reduce=op) return torch.stack([storage.row(), storage.col()], dim=0), storage.value()