Skip to content

Commit

Permalink
fix triton test workflow
Browse files Browse the repository at this point in the history
  • Loading branch information
phtvo committed Dec 29, 2023
1 parent 1f9aed1 commit 408b6ad
Showing 1 changed file with 17 additions and 16 deletions.
33 changes: 17 additions & 16 deletions .github/workflows/run_triton_test.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,11 @@ name: Run triton test
on:
push:
branches: [ master ]
paths:
- 'clarifai/models/**'
pull_request:
paths:
- 'clarifai/models/**'

jobs:
triton-model-test:
Expand All @@ -13,20 +17,17 @@ jobs:
image: nvcr.io/nvidia/tritonserver:23.10-py3

steps:
- name: Install dependencies
- name: Install dependencies
run: |
python -m pip install --upgrade pip
pip install -r requirements.txt
pip install -r tests/requirements.txt
- name: start triton
run: |
python -m pip install --upgrade pip
pip install -r requirements.txt
pip install -r tests/requirements.txt
- name: start triton
run: |
tritonserver --model-repository ./tests/dummy_triton_models --exit-on-error false &
background: true

- name: Wait for Triton to Initialize
run: sleep 10

- name: Start Model Tests
run: |
python3 -m pytest tests/dummy_triton_models/_test_all_dummies.py -s
tritonserver --model-repository ./tests/dummy_triton_models --exit-on-error false &
background: true
- name: Wait for Triton to Initialize
run: sleep 10
- name: Start Model Tests
run: |
python3 -m pytest tests/dummy_triton_models/_test_all_dummies.py -s

0 comments on commit 408b6ad

Please sign in to comment.