From 93e620d8038515e1528629782525bfe2758c92de Mon Sep 17 00:00:00 2001 From: Zach Kimberg Date: Thu, 2 Aug 2018 18:09:20 -0700 Subject: [PATCH] Fix flaky tests for test_squared_hinge_loss --- tests/python/unittest/test_loss.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/python/unittest/test_loss.py b/tests/python/unittest/test_loss.py index 3c147bc4c723..3fb9d784971b 100644 --- a/tests/python/unittest/test_loss.py +++ b/tests/python/unittest/test_loss.py @@ -310,7 +310,7 @@ def test_hinge_loss(): assert mod.score(data_iter, eval_metric=mx.metric.Loss())[0][1] < 0.05 -@with_seed(1234) +@with_seed() def test_squared_hinge_loss(): N = 20 data = mx.random.uniform(-1, 1, shape=(N, 10))