@@ -118,7 +118,7 @@ def is_service_exposed(self) -> bool:
118118 """
119119 labels = self .inference_service .labels
120120
121- if self .deployment_mode == KServeDeploymentType .RAW_DEPLOYMENT :
121+ if self .deployment_mode in KServeDeploymentType .RAW_DEPLOYMENT_MODES :
122122 if isinstance (self .inference_service , InferenceGraph ):
123123 # For InferenceGraph, the logic is similar as in Serverless. Only the label is different.
124124 return not (labels and labels .get (Labels .Kserve .NETWORKING_KSERVE_IO ) == "cluster-local" )
@@ -310,7 +310,7 @@ def generate_command(
310310
311311 elif self .protocol == "grpc" :
312312 cmd_exec = "grpcurl -connect-timeout 10 "
313- if self .deployment_mode == KServeDeploymentType .RAW_DEPLOYMENT :
313+ if self .deployment_mode in KServeDeploymentType .RAW_DEPLOYMENT_MODES :
314314 cmd_exec += " --plaintext "
315315
316316 else :
@@ -536,10 +536,9 @@ def get_target_port(self, svc: Service) -> int:
536536 and port .protocol .lower () == svc_protocol .lower ()
537537 and port .name == self .protocol
538538 ) or (
539- self .deployment_mode
540- in (
541- KServeDeploymentType .RAW_DEPLOYMENT ,
542- KServeDeploymentType .SERVERLESS ,
539+ (
540+ self .deployment_mode in KServeDeploymentType .RAW_DEPLOYMENT_MODES
541+ or self .deployment_mode == KServeDeploymentType .SERVERLESS
543542 )
544543 and port .protocol .lower () == svc_protocol .lower ()
545544 ):
@@ -679,15 +678,18 @@ def create_isvc(
679678 _annotations = {Annotations .KserveIo .DEPLOYMENT_MODE : deployment_mode }
680679
681680 # model mesh auth is set in ServingRuntime
682- if enable_auth and deployment_mode in {KServeDeploymentType .SERVERLESS , KServeDeploymentType .RAW_DEPLOYMENT }:
681+ if enable_auth and (
682+ deployment_mode == KServeDeploymentType .SERVERLESS
683+ or deployment_mode in KServeDeploymentType .RAW_DEPLOYMENT_MODES
684+ ):
683685 _annotations [Annotations .KserveAuth .SECURITY ] = "true"
684686
685687 # default to True if deployment_mode is Serverless (default behavior of Serverless) if was not provided by the user
686688 # model mesh external route is set in ServingRuntime
687689 if external_route is None and deployment_mode == KServeDeploymentType .SERVERLESS :
688690 external_route = True
689691
690- if external_route and deployment_mode == KServeDeploymentType .RAW_DEPLOYMENT :
692+ if external_route and deployment_mode in KServeDeploymentType .RAW_DEPLOYMENT_MODES :
691693 labels [Labels .Kserve .NETWORKING_KSERVE_IO ] = Labels .Kserve .EXPOSED
692694
693695 if deployment_mode == KServeDeploymentType .SERVERLESS and external_route is False :
0 commit comments