diff --git a/charms/kserve-controller/src/default-custom-images.json b/charms/kserve-controller/src/default-custom-images.json index 08554aa..bfa1b63 100644 --- a/charms/kserve-controller/src/default-custom-images.json +++ b/charms/kserve-controller/src/default-custom-images.json @@ -14,7 +14,7 @@ "serving_runtimes__tensorflow_serving": "tensorflow/serving:2.6.2", "serving_runtimes__torchserve": "pytorch/torchserve-kfs:0.8.2", "serving_runtimes__tritonserver": "nvcr.io/nvidia/tritonserver:23.05-py3", - "serving_runtimes__xgbserver": "kserve/xgbserver:v0.11.1" + "serving_runtimes__xgbserver": "charmedkubeflow/xgbserver:0.11.2-d1246b8" } diff --git a/charms/kserve-controller/tests/integration/test_charm.py b/charms/kserve-controller/tests/integration/test_charm.py index de911a9..dbbdb08 100644 --- a/charms/kserve-controller/tests/integration/test_charm.py +++ b/charms/kserve-controller/tests/integration/test_charm.py @@ -246,6 +246,7 @@ def create_namespace(): "./tests/integration/lgbserver.yaml", "./tests/integration/pmml-server.yaml", "./tests/integration/paddleserver-resnet.yaml", + "./tests/integration/xgbserver.yaml", ], ) def test_inference_service_raw_deployment( diff --git a/charms/kserve-controller/tests/integration/xgbserver.yaml b/charms/kserve-controller/tests/integration/xgbserver.yaml new file mode 100644 index 0000000..589929a --- /dev/null +++ b/charms/kserve-controller/tests/integration/xgbserver.yaml @@ -0,0 +1,19 @@ +apiVersion: "serving.kserve.io/v1beta1" +kind: "InferenceService" +metadata: + name: "xgboost-v2-iris" +spec: + predictor: + model: + modelFormat: + name: xgboost + protocolVersion: v2 + runtime: kserve-xgbserver + storageUri: "gs://kfserving-examples/models/xgboost/iris" + resources: + limits: + cpu: 1 + memory: 500Mi + requests: + cpu: 100m + memory: 250Mi