Integrate rock for xgbserver v0.11.2

canonical · Mar 7, 2024 · ade6e0a · ade6e0a
1 parent 5341043
commit ade6e0a
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 1 deletion.
diff --git 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
@@ -201,6 +201,7 @@ def create_namespace():
         "./tests/integration/sklearn-iris.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
@@ -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