Skip to content

Commit a9ea2d5

Browse files
committed
Fixes unit tests
1 parent ea04433 commit a9ea2d5

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

tests/unitary/with_extras/aqua/test_common_utils.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
# Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
66

77
import pytest
8+
89
from ads.aqua.common.utils import get_preferred_compatible_family
910

1011

@@ -14,23 +15,23 @@ class TestCommonUtils:
1415
[
1516
(
1617
{"odsc-vllm-serving", "odsc-vllm-serving-v1"},
17-
"odsc-vllm-serving-v1",
18+
"odsc-vllm-serving-openai",
1819
),
1920
(
2021
{"odsc-vllm-serving", "odsc-vllm-serving-llama4"},
21-
"odsc-vllm-serving-llama4",
22+
"odsc-vllm-serving-openai",
2223
),
2324
(
2425
{"odsc-vllm-serving-v1", "odsc-vllm-serving-llama4"},
25-
"odsc-vllm-serving-llama4",
26+
"odsc-vllm-serving-openai",
2627
),
2728
(
2829
{
2930
"odsc-vllm-serving",
3031
"odsc-vllm-serving-v1",
3132
"odsc-vllm-serving-llama4",
3233
},
33-
"odsc-vllm-serving-llama4",
34+
"odsc-vllm-serving-openai",
3435
),
3536
({"odsc-tgi-serving", "odsc-vllm-serving"}, None),
3637
({"non-existing-one", "odsc-tgi-serving"}, None),

0 commit comments

Comments
 (0)