From 49598975465cf8d0ef34d22ac553bbd8d1b50896 Mon Sep 17 00:00:00 2001 From: Ryan McCormick Date: Thu, 16 Jan 2025 11:53:15 -0800 Subject: [PATCH 1/2] Increase test timeout for test stability for now --- tests/test_e2e.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/tests/test_e2e.py b/tests/test_e2e.py index d48aa39..daeb954 100644 --- a/tests/test_e2e.py +++ b/tests/test_e2e.py @@ -52,7 +52,9 @@ class TestE2E: ), ], ) - @pytest.mark.timeout(600) + # Give ample 30min timeout for now as this test will currently download + # models from huggingface as well, and network speed is intermittent. + @pytest.mark.timeout(1800) def test_tensorrtllm_e2e(self, llm_server, protocol): # NOTE: TRTLLM test models will be passed by the testing infrastructure. # Only a single model will be passed per test to enable tests to run concurrently. @@ -82,7 +84,9 @@ def test_tensorrtllm_e2e(self, llm_server, protocol): ), ], ) - @pytest.mark.timeout(900) + # Give ample 30min timeout for now as this test will currently download + # models from huggingface as well, and network speed is intermittent. + @pytest.mark.timeout(1800) def test_vllm_e2e(self, llm_server, protocol): # NOTE: VLLM test models will be passed by the testing infrastructure. # Only a single model will be passed per test to enable tests to run concurrently. From 2d4cc8e9484452bc3e6fe5b3a45d38d0f4b9861f Mon Sep 17 00:00:00 2001 From: Ryan McCormick Date: Thu, 16 Jan 2025 12:06:39 -0800 Subject: [PATCH 2/2] Update copyright --- tests/test_e2e.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_e2e.py b/tests/test_e2e.py index daeb954..95edc2c 100644 --- a/tests/test_e2e.py +++ b/tests/test_e2e.py @@ -1,4 +1,4 @@ -# Copyright 2024, NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# Copyright 2024-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved. # # Redistribution and use in source and binary forms, with or without # modification, are permitted provided that the following conditions