Skip to content

Commit 46438e2

Browse files
committed
update: test_d_adapt_reset
1 parent cdebfd6 commit 46438e2

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/test_optimizers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -296,7 +296,7 @@ def test_reset(optimizer_config):
296296
@pytest.mark.parametrize('sparse_gradient', [False, True])
297297
@pytest.mark.parametrize('optimizer_name', ['DAdaptAdaGrad', 'DAdaptAdam', 'DAdaptSGD'])
298298
def test_d_adapt_reset(require_gradient, sparse_gradient, optimizer_name):
299-
param = simple_sparse_parameter(require_gradient) if sparse_gradient else simple_parameter(require_gradient)
299+
param = simple_sparse_parameter(require_gradient)[1] if sparse_gradient else simple_parameter(require_gradient)
300300
if not require_gradient:
301301
param.grad = None
302302

0 commit comments

Comments
 (0)