Skip to content

GitLab

Commit 8a8eb34f authored by hubertlu-tw's avatar hubertlu-tw
Browse files

Skip a flaky unit test

parent 975a0e53
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
tests/L0/run_mlp/test_mlp.py
View file @ 8a8eb34f
...@@ -134,6 +134,7 @@ class TestMLP(unittest.TestCase): ...@@ -134,6 +134,7 @@ class TestMLP(unittest.TestCase):
ref_mlp[0].bias.grad.detach().cpu().numpy(), ref_mlp[0].bias.grad.detach().cpu().numpy(),
atol=1e-7, rtol=1e-5) atol=1e-7, rtol=1e-5)
@skipFlakyTest
def test_no_grad(self): def test_no_grad(self):
mlp = MLP(mlp_sizes).cuda() mlp = MLP(mlp_sizes).cuda()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment