diff --git a/tests/python/unittest/test_optimizer.py b/tests/python/unittest/test_optimizer.py index fdf7d279d9cc..669561b9a45d 100644 --- a/tests/python/unittest/test_optimizer.py +++ b/tests/python/unittest/test_optimizer.py @@ -442,7 +442,7 @@ def update(self, index, weight, grad, state): tmp = weight32.astype(weight.dtype) tmp.copyto(weight) -@with_seed(0) +@with_seed() def test_nag(): opt1 = PyNAG opt2 = mx.optimizer.NAG