diff --git a/test/test_sparse.py b/test/test_sparse.py index ba7b5d527a28..e365c94aae99 100644 --- a/test/test_sparse.py +++ b/test/test_sparse.py @@ -1107,7 +1107,7 @@ def test_shape(di, dj, dk, nnz): # Test code from issue https://github.com/pytorch/pytorch/issues/45113 batch_size, input_size, hidden_size = 5, 3, 7 - + # Create coalesced sparse tensor as in the issue weight = torch.randn(hidden_size, input_size).to_sparse() self.assertTrue(weight.is_coalesced())