From 3289c1207d238fada989ca1dcd3b948befb870be Mon Sep 17 00:00:00 2001 From: Yineng Zhang Date: Thu, 3 Apr 2025 17:07:38 -0700 Subject: [PATCH] Update the retry count (#5051) --- .github/workflows/pr-test.yml | 2 +- python/sglang/test/test_utils.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/pr-test.yml b/.github/workflows/pr-test.yml index b26cfecdc..dd6370091 100644 --- a/.github/workflows/pr-test.yml +++ b/.github/workflows/pr-test.yml @@ -68,7 +68,7 @@ jobs: bash scripts/ci_install_dependency.sh - name: Run test - timeout-minutes: 30 + timeout-minutes: 40 run: | cd test/srt python3 run_suite.py --suite per-commit --auto-partition-id ${{ matrix.part }} --auto-partition-size 7 diff --git a/python/sglang/test/test_utils.py b/python/sglang/test/test_utils.py index 941616681..fe876c960 100644 --- a/python/sglang/test/test_utils.py +++ b/python/sglang/test/test_utils.py @@ -1014,7 +1014,7 @@ def run_logprob_check(self: unittest.TestCase, arg: Tuple): class CustomTestCase(unittest.TestCase): def _callTestMethod(self, method): max_retry = int( - os.environ.get("SGLANG_TEST_MAX_RETRY", "2" if is_in_ci() else "0") + os.environ.get("SGLANG_TEST_MAX_RETRY", "1" if is_in_ci() else "0") ) retry( lambda: super(CustomTestCase, self)._callTestMethod(method),