[Test] Fix flaky LocalRunner test due to a small timeout (apache#9181)

ylc · Jan 7, 2022 · 95fa570 · 95fa570
1 parent e19fa8b
commit 95fa570
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/python/tvm/meta_schedule/runner/local_runner.py b/python/tvm/meta_schedule/runner/local_runner.py
@@ -175,7 +175,7 @@ def default_cleanup() -> None:
 
     def __init__(
         self,
-        timeout_sec: float,
+        timeout_sec: float = 30,
         evaluator_config: Optional[EvaluatorConfig] = None,
         cooldown_sec: float = 0.0,
         alloc_repeat: int = 1,

diff --git a/tests/python/unittest/test_meta_schedule_runner.py b/tests/python/unittest/test_meta_schedule_runner.py
@@ -507,7 +507,7 @@ def exception_session_creator(  # pylint: disable=unused-variable
 
 
 def test_meta_schedule_local_runner_exception():
-    """Test meta schedule Local Runner time out"""
+    """Test meta schedule Local Runner exception"""
     mod = MatmulModule
     builder = LocalBuilder()
     (builder_result,) = builder.build([BuilderInput(mod, Target("llvm"))])
@@ -541,7 +541,6 @@ def timeout_session_creator(  # pylint: disable=unused-variable
     )
 
     runner = LocalRunner(
-        timeout_sec=1,
         evaluator_config=evaluator_config,
         initializer=initializer,
         f_alloc_argument="meta_schedule.runner.test_exception",