Skip to content

Commit 215898f

Browse files
committed
fix: fix annotation
1 parent d9d21b8 commit 215898f

File tree

1 file changed

+2
-8
lines changed

1 file changed

+2
-8
lines changed

tests/model_serving/model_server/components/kserve_dsc_deployment_mode/test_kserve_dsc_default_deployment_mode.py

Lines changed: 2 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -50,10 +50,7 @@ def test_isvc_contains_serverless_deployment_mode(
5050
ovms_inference_service,
5151
):
5252
"""Verify that default deployment mode is set to serverless in inference service."""
53-
assert (
54-
ovms_inference_service.instance.metadata.annotations[Annotations.KserveIo.DEPLOYMENT_MODE]
55-
== KServeDeploymentType.SERVERLESS
56-
)
53+
assert ovms_inference_service.instance.status.deploymentMode == KServeDeploymentType.SERVERLESS
5754

5855
def test_kserve_dsc_serverless_default_deployment_mode(
5956
self, default_deployment_mode_in_dsc, ovms_inference_service
@@ -83,10 +80,7 @@ def test_isvc_on_dsc_default_deployment_mode_change_to_raw(
8380
ovms_inference_service,
8481
):
8582
"""Verify that Serverless isvc not changed after dsc default deployment mode is changed to raw"""
86-
assert (
87-
ovms_inference_service.instance.metadata.annotations[Annotations.KserveIo.DEPLOYMENT_MODE]
88-
== KServeDeploymentType.SERVERLESS
89-
)
83+
assert ovms_inference_service.instance.status.deploymentMode == KServeDeploymentType.SERVERLESS
9084

9185
@pytest.mark.parametrize(
9286
"patched_default_deployment_mode_in_dsc",

0 commit comments

Comments
 (0)