Disable pytest-timeout for now. (#8348)

2022-10-17 23:06:10 +08:00
parent fcddbc9264
commit 2176e511fc
13 changed files with 107 additions and 58 deletions
--- a/tests/python/test_demos.py
+++ b/tests/python/test_demos.py
@@ -1,10 +1,13 @@
 import os
 import subprocess
-import pytest
-import testing as tm
 import sys

-pytestmark = pytest.mark.timeout(30)
+import pytest
+import testing as tm
+
+from xgboost import testing
+
+pytestmark = testing.timeout(30)

 ROOT_DIR = tm.PROJECT_ROOT
 DEMO_DIR = os.path.join(ROOT_DIR, 'demo')