Increase test timeout for test_handler_traceback_logging. (#3113)

This is to handle test_utils.torchserve_cleanup() taking longer in Docker.
pytorch · Apr 24, 2024 · 3d23fc3 · 3d23fc3
1 parent bebab5a
commit 3d23fc3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/pytest/test_handler_traceback_logging.py b/test/pytest/test_handler_traceback_logging.py
@@ -156,7 +156,7 @@ def register_model(mar_file_path, model_store, torchserve):
     test_utils.unregister_model(model_name)
 
 
-@pytest.mark.timeout(60)
+@pytest.mark.timeout(120)
 def test_handler_traceback_logging(model_name):
     """
     Full circle test with torchserve