Insert retry

chainer · Aug 26, 2017 · 8b5bc2f · 8b5bc2f
1 parent 919f584
commit 8b5bc2f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/chainer_tests/test_gradient_check.py b/tests/chainer_tests/test_gradient_check.py
@@ -329,6 +329,7 @@ def backward(self, inputs, grads):
 }))
 class TestCheckBackward(unittest.TestCase):
 
+    @condition.retry(3)
     def test_multiple_output(self):
         x1 = numpy.array([1], dtype='f')
         x2 = numpy.array([1], dtype='f')
@@ -342,6 +343,7 @@ def f(x, y):
 
         gradient_check.check_backward(f, (x1, x2), (g1, g2), dtype=self.dtype)
 
+    @condition.retry(3)
     def test_no_grads_for_not_float(self):
         x1 = numpy.array([1], dtype='f')
         x2 = numpy.array([0, 1], dtype='i')  # grad check for this is skipped
@@ -351,8 +353,9 @@ def f(x, y):
             s = Ident()(x)
             return s,
 
-        gradient_check.check_backward(f, (x1, x2), g1, dtype='d')
+        gradient_check.check_backward(f, (x1, x2), g1, dtype=self.dtype)
 
+    @condition.retry(3)
     def test_no_grads_option(self):
         x1 = numpy.array([1], dtype='f')
         x2 = numpy.array([1], dtype='f')  # grad check for this is skipped