refactor: grad

Author: kozistr

tests/test_sparse_gradient.py

@@ -29,8 +29,7 @@
 @pytest.mark.parametrize('no_sparse_optimizer', NO_SPARSE_OPTIMIZERS)
 def test_sparse_not_supported(no_sparse_optimizer):
     param = torch.randn(1, 1).to_sparse(1).requires_grad_(True)
-    grad = torch.randn(1, 1).to_sparse(1)
-    param.grad = grad
+    param.grad = torch.randn(1, 1).to_sparse(1)
 
     with pytest.raises(RuntimeError):
         optimizer = load_optimizers(optimizer=no_sparse_optimizer)([param])
@@ -41,8 +40,7 @@ def test_sparse_not_supported(no_sparse_optimizer):
 @pytest.mark.parametrize('sparse_optimizer', SPARSE_OPTIMIZERS)
 def test_sparse_supported(sparse_optimizer):
     param = torch.randn(1, 1).to_sparse(1).requires_grad_(True)
-    grad = torch.randn(1, 1).to_sparse(1)
-    param.grad = grad
+    param.grad = torch.randn(1, 1).to_sparse(1)
 
     optimizer = load_optimizers(optimizer=sparse_optimizer)([param], momentum=0.0)
     optimizer.zero_grad()