From 17a92cb664e0a4453afb39518a12eba18bcb1723 Mon Sep 17 00:00:00 2001 From: Julius von Kohout <45896133+juliusvonkohout@users.noreply.github.com> Date: Thu, 5 Oct 2023 17:18:06 +0200 Subject: [PATCH] Update kserve.py --- tests/e2e/utils/kserve.py | 9 +-------- 1 file changed, 1 insertion(+), 8 deletions(-) diff --git a/tests/e2e/utils/kserve.py b/tests/e2e/utils/kserve.py index d62ca6b50d..a69c097184 100644 --- a/tests/e2e/utils/kserve.py +++ b/tests/e2e/utils/kserve.py @@ -1,15 +1,8 @@ from kfp import components - def create_serving_task(model_name, model_namespace, tfjob_op, model_volume_op): - api_version = 'serving.kserve.io/v1beta1' serving_component_url = 'https://raw.githubusercontent.com/kubeflow/pipelines/master/components/kserve/component.yaml' - - # Uncomment the following two lines if you are using KFServing v0.6.x or v0.5.x - # api_version = 'serving.kubeflow.org/v1beta1' - # serving_component_url = 'https://raw.githubusercontent.com/kubeflow/pipelines/master/components/kubeflow/kfserving/component.yaml' - inference_service = ''' apiVersion: "{}" kind: "InferenceService" @@ -25,4 +18,4 @@ def create_serving_task(model_name, model_namespace, tfjob_op, model_volume_op): '''.format(api_version, model_name, model_namespace, str(model_volume_op.outputs["name"])) serving_launcher_op = components.load_component_from_url(serving_component_url) - serving_launcher_op(action="apply", inferenceservice_yaml=inference_service).after(tfjob_op) \ No newline at end of file + serving_launcher_op(action="apply", inferenceservice_yaml=inference_service).after(tfjob_op)