Trying to find a clue about the flaky test of (#10995)

test_regression_class_build_fn.
okly366 · Aug 28, 2018 · c6beaa5 · c6beaa5
1 parent be6d829
commit c6beaa5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -45,6 +45,7 @@
                     'pytest-pep8',
                     'pytest-xdist',
                     'pytest-cov',
+                    'pytest-timeout',
                     'pandas',
                     'requests'],
       },

diff --git a/tests/keras/wrappers/scikit_learn_test.py b/tests/keras/wrappers/scikit_learn_test.py
@@ -131,6 +131,7 @@ def test_regression_build_fn():
     assert_regression_works(reg)
 
 
+@pytest.mark.timeout(50)
 def test_regression_class_build_fn():
     class ClassBuildFnReg(object):