Add annotation check for the deployment mode (kubeflow#2338)

Signed-off-by: Dan Sun <dsun20@bloomberg.net>
magdalenakuhn17 · Jul 16, 2022 · 133eceb · 133eceb
1 parent a6ed8e4
commit 133eceb
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/pkg/apis/serving/v1beta1/inference_service_defaults.go b/pkg/apis/serving/v1beta1/inference_service_defaults.go
@@ -80,9 +80,11 @@ func (isvc *InferenceService) Default() {
 }
 
 func (isvc *InferenceService) DefaultInferenceService(config *InferenceServicesConfig, deployConfig *DeployConfig) {
-	if deployConfig.DefaultDeploymentMode == string(constants.ModelMeshDeployment) ||
-		deployConfig.DefaultDeploymentMode == string(constants.RawDeployment) {
-		isvc.ObjectMeta.Annotations[constants.DeploymentMode] = deployConfig.DefaultDeploymentMode
+	if _, ok := isvc.ObjectMeta.Annotations[constants.DeploymentMode]; !ok {
+		if deployConfig.DefaultDeploymentMode == string(constants.ModelMeshDeployment) ||
+			deployConfig.DefaultDeploymentMode == string(constants.RawDeployment) {
+			isvc.ObjectMeta.Annotations[constants.DeploymentMode] = deployConfig.DefaultDeploymentMode
+		}
 	}
 	components := []Component{isvc.Spec.Transformer, isvc.Spec.Explainer}
 	deploymentMode, ok := isvc.ObjectMeta.Annotations[constants.DeploymentMode]