Skip to content

GitLab

Unverified Commit 3289c120 authored by Yineng Zhang's avatar Yineng Zhang Committed by GitHub
Browse files

Update the retry count (#5051)

parent 69df9761
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
.github/workflows/pr-test.yml
View file @ 3289c120
...@@ -68,7 +68,7 @@ jobs: ...@@ -68,7 +68,7 @@ jobs:
bash scripts/ci_install_dependency.sh bash scripts/ci_install_dependency.sh
- name: Run test - name: Run test
timeout-minutes: 30 timeout-minutes: 40
run: | run: |
cd test/srt cd test/srt
python3 run_suite.py --suite per-commit --auto-partition-id ${{ matrix.part }} --auto-partition-size 7 python3 run_suite.py --suite per-commit --auto-partition-id ${{ matrix.part }} --auto-partition-size 7
......
python/sglang/test/test_utils.py
View file @ 3289c120
...@@ -1014,7 +1014,7 @@ def run_logprob_check(self: unittest.TestCase, arg: Tuple): ...@@ -1014,7 +1014,7 @@ def run_logprob_check(self: unittest.TestCase, arg: Tuple):
class CustomTestCase(unittest.TestCase): class CustomTestCase(unittest.TestCase):
def _callTestMethod(self, method): def _callTestMethod(self, method):
max_retry = int( 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( retry(
lambda: super(CustomTestCase, self)._callTestMethod(method), lambda: super(CustomTestCase, self)._callTestMethod(method),
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment