Fix runtime spec for test

Signed-off-by: Rafael Vasquez <[email protected]>
kserve · Mar 13, 2024 · 910bb0c · 910bb0c
1 parent ad93a4e
commit 910bb0c
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/fvt/predictor/predictor_test.go b/fvt/predictor/predictor_test.go
@@ -1052,6 +1052,11 @@ var _ = Describe("Invalid Predictors", func() {
 				// modify the object with an unrecognized model type
 				SetString(predictorObject, "invalidModelType", "spec", "modelType", "name")
 
+				// remove runtime field for Triton-FIL predictors for this test
+				if GetString(predictorObject, "spec", "runtime", "name") == "triton-2.x" {
+					SetString(predictorObject, "", "spec", "runtime", "name")
+				}
+
 				obj := CreatePredictorAndWaitAndExpectFailed(predictorObject)
 
 				By("Verifying the predictor")