diff --git a/.github/workflows/test_openvino_examples.yml b/.github/workflows/test_openvino_examples.yml index a1b7f09b26..61b411c967 100644 --- a/.github/workflows/test_openvino_examples.yml +++ b/.github/workflows/test_openvino_examples.yml @@ -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' @@ -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 @@ -36,7 +35,7 @@ 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 @@ -44,4 +43,4 @@ jobs: - name: Test examples run: | - python examples/openvino/test_examples.py \ No newline at end of file + python -m pytest examples/openvino/test_examples.py \ No newline at end of file diff --git a/examples/openvino/test_examples.py b/examples/openvino/test_examples.py index a59b8b5690..d3993d5b78 100644 --- a/examples/openvino/test_examples.py +++ b/examples/openvino/test_examples.py @@ -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