Fix flaky tests for test_squared_hinge_loss (apache#12017)

XinYao1994 · Aug 4, 2018 · ee24e16 · ee24e16
1 parent c0c508e
commit ee24e16
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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))