Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
Fix flaky tests for test_squared_hinge_loss
Browse files Browse the repository at this point in the history
  • Loading branch information
zachgk committed Aug 3, 2018
1 parent bcfab3a commit 8abe011
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion tests/python/unittest/test_loss.py
Original file line number Diff line number Diff line change
Expand Up @@ -310,7 +310,9 @@ def test_hinge_loss():
assert mod.score(data_iter, eval_metric=mx.metric.Loss())[0][1] < 0.05


@with_seed(1234)
# @zachgk: Getting ride of fixed seed as flakiness could not be reproduced,
# tracked at https://github.com/apache/incubator-mxnet/issues/11702
@with_seed()
def test_squared_hinge_loss():
N = 20
data = mx.random.uniform(-1, 1, shape=(N, 10))
Expand Down

0 comments on commit 8abe011

Please sign in to comment.