diff --git a/tests/python/unittest/test_optimizer.py b/tests/python/unittest/test_optimizer.py index fba10fb522a2..a5b3d4047df9 100644 --- a/tests/python/unittest/test_optimizer.py +++ b/tests/python/unittest/test_optimizer.py @@ -1034,6 +1034,8 @@ def test_adagrad(): if wd_option.get('wd', 0.0) == 0.0: compare_optimizer(opt1(**kwarg), opt2(**kwarg), shape, dtype, w_stype='row_sparse', g_stype='row_sparse') + compare_optimizer(opt1(**kwarg), opt2(**kwarg), shape, dtype, + g_stype='row_sparse')