diff --git a/.github/workflows/test_openvino_examples.yml b/.github/workflows/test_openvino_examples.yml index a1b7f09b26..2c57264551 100644 --- a/.github/workflows/test_openvino_examples.yml +++ b/.github/workflows/test_openvino_examples.yml @@ -44,4 +44,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