diff --git a/machine-learning/test_main.py b/machine-learning/test_main.py index d5ef5c20da..9aa611622c 100644 --- a/machine-learning/test_main.py +++ b/machine-learning/test_main.py @@ -286,7 +286,7 @@ class TestOrtSession: def test_sets_provider_options_kwarg(self) -> None: session = OrtSession( "ViT-B-32__openai", - providers=["OpenVINOExecutionProvider", "CPUExecutionProvider"], + providers=["CUDAExecutionProvider", "CPUExecutionProvider"], provider_options=[], )