diff --git a/tests/python/unittest/test_loss.py b/tests/python/unittest/test_loss.py index 8d5b86341a88..cb55cda9250f 100644 --- a/tests/python/unittest/test_loss.py +++ b/tests/python/unittest/test_loss.py @@ -127,7 +127,7 @@ def test_logistic_loss_equal_bce(): assert_almost_equal(loss_binary(data, label).asnumpy(), loss_bce(data, label).asnumpy()) assert_almost_equal(loss_signed(data, 2 * label - 1).asnumpy(), loss_bce(data, label).asnumpy()) -@with_seed(1234) +@with_seed() def test_kl_loss(): N = 20 data = mx.random.uniform(-1, 1, shape=(N, 10))