From eef9d32b27e7e7234d8bb4f6751896e37f491ec5 Mon Sep 17 00:00:00 2001 From: Haibin Lin Date: Wed, 1 Aug 2018 22:51:06 -0700 Subject: [PATCH] Update test_loss.py --- tests/python/unittest/test_loss.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/python/unittest/test_loss.py b/tests/python/unittest/test_loss.py index 8d5b86341a88..fea8e5f26693 100644 --- a/tests/python/unittest/test_loss.py +++ b/tests/python/unittest/test_loss.py @@ -144,7 +144,7 @@ def test_kl_loss(): assert mod.score(data_iter, eval_metric=mx.metric.Loss())[0][1] < 0.05 -@with_seed(1234) +@with_seed() def test_l2_loss(): N = 20 data = mx.random.uniform(-1, 1, shape=(N, 10)) @@ -162,7 +162,7 @@ def test_l2_loss(): assert mod.score(data_iter, eval_metric=mx.metric.Loss())[0][1] < 0.05 -@with_seed(1234) +@with_seed() def test_l1_loss(): N = 20 data = mx.random.uniform(-1, 1, shape=(N, 10))