diff --git a/operator/utils/utils.go b/operator/utils/utils.go index 9e0ba5672b..2311764de5 100644 --- a/operator/utils/utils.go +++ b/operator/utils/utils.go @@ -54,7 +54,7 @@ func GetPredictionPath(mlDep *machinelearningv1.SeldonDeployment) string { if protocol == machinelearningv1.ProtocolTensorflow { // This will be updated as part of https://github.com/SeldonIO/seldon-core/issues/1611 - return "/v1/models/" + mlDep.Spec.Predictors[0].Graph.Name + "/:predict" + return "/v1/models/" + mlDep.Spec.Predictors[0].Graph.Name + ":predict" } else if protocol == machinelearningv1.ProtocolKfserving { return "/v2/models/" + mlDep.Spec.Predictors[0].Graph.Name + "/infer" } else { diff --git a/operator/utils/utils_test.go b/operator/utils/utils_test.go index 48985edbb6..b260f773eb 100644 --- a/operator/utils/utils_test.go +++ b/operator/utils/utils_test.go @@ -182,5 +182,5 @@ func TestTensorflowPredictionPath(t *testing.T) { } p := GetPredictionPath(sdep) - g.Expect(p).To(Equal("/v1/models/classifier/:predict")) + g.Expect(p).To(Equal("/v1/models/classifier:predict")) }