Skip to content

Commit

Permalink
Udpate model ID for OpenVINO example test (#616)
Browse files Browse the repository at this point in the history
* udpate model test

* add pytest install
  • Loading branch information
echarlaix authored Mar 19, 2024
1 parent 9813f90 commit d2f9fdb
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 5 deletions.
7 changes: 3 additions & 4 deletions .github/workflows/test_openvino_examples.yml
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@ name: OpenVINO - Examples Test
on:
workflow_dispatch:
schedule:
- cron: '14 3 * * 1' # run weekly: every Monday at 3:14
- cron: 0 1 * * 1 # run weekly: every Monday at 1am
push:
paths:
- '.github/workflows/test_openvino_examples.yml'
Expand All @@ -13,7 +13,6 @@ on:
- '.github/workflows/test_openvino_examples.yml'
- 'examples/openvino/*'


concurrency:
group: ${{ github.workflow }}-${{ github.head_ref || github.run_id }}
cancel-in-progress: true
Expand All @@ -36,12 +35,12 @@ jobs:

- name: Install dependencies
run: |
pip install optimum[openvino] jstyleson nncf
pip install optimum[openvino] jstyleson nncf pytest
pip install -r examples/openvino/audio-classification/requirements.txt
pip install -r examples/openvino/image-classification/requirements.txt
pip install -r examples/openvino/question-answering/requirements.txt
pip install -r examples/openvino/text-classification/requirements.txt
- name: Test examples
run: |
python examples/openvino/test_examples.py
python -m pytest examples/openvino/test_examples.py
2 changes: 1 addition & 1 deletion examples/openvino/test_examples.py
Original file line number Diff line number Diff line change
Expand Up @@ -73,7 +73,7 @@ def test_image_classification(self):
with tempfile.TemporaryDirectory() as tmp_dir:
test_args = f"""
run_image_classification.py
--model_name_or_path nateraw/vit-base-beans
--model_name_or_path hf-internal-testing/tiny-random-ViTModel
--dataset_name beans
--max_train_samples 10
--max_eval_samples 2
Expand Down

0 comments on commit d2f9fdb

Please sign in to comment.