From e60b995c3e175e49e10c29e32ea6a860f87759f4 Mon Sep 17 00:00:00 2001
From: Aleksander <pejcic@adobe.com>
Date: Wed, 15 Oct 2025 09:39:38 -0700
Subject: [PATCH] Fix openvino tests

---
 machine-learning/test_main.py | 10 ++--------
 1 file changed, 2 insertions(+), 8 deletions(-)

diff --git a/machine-learning/test_main.py b/machine-learning/test_main.py
index 20f9ec9ac9..d5ef5c20da 100644
--- a/machine-learning/test_main.py
+++ b/machine-learning/test_main.py
@@ -201,13 +201,6 @@ class TestOrtSession:
 
         assert session.providers == self.OV_EP
 
-    @pytest.mark.ov_device_ids(["CPU"])
-    @pytest.mark.providers(OV_EP)
-    def test_avoids_openvino_if_gpu_not_available(self, providers: list[str], ov_device_ids: list[str]) -> None:
-        session = OrtSession("ViT-B-32__openai")
-
-        assert session.providers == self.CPU_EP
-
     @pytest.mark.providers(CUDA_EP_OUT_OF_ORDER)
     def test_sets_providers_in_correct_order(self, providers: list[str]) -> None:
         session = OrtSession("ViT-B-32__openai")
@@ -248,7 +241,8 @@ class TestOrtSession:
             {"arena_extend_strategy": "kSameAsRequested"},
         ]
 
-    def test_sets_provider_options_for_openvino(self) -> None:
+    @pytest.mark.ov_device_ids(["GPU.0", "GPU.1", "CPU"])
+    def test_sets_provider_options_for_openvino(self, ov_device_ids: list[str]) -> None:
         model_path = "/cache/ViT-B-32__openai/textual/model.onnx"
         os.environ["MACHINE_LEARNING_DEVICE_ID"] = "1"