Integrate rock for lgbserver v0.11.2 (#221)

canonical · Mar 8, 2024 · caa8f1c · caa8f1c
1 parent 13c3ce4
commit caa8f1c
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 1 deletion.
diff --git a/charms/kserve-controller/src/default-custom-images.json b/charms/kserve-controller/src/default-custom-images.json
@@ -6,7 +6,7 @@
     "configmap__logger": "kserve/agent:v0.11.1",
     "configmap__router": "kserve/router:v0.11.1",
     "configmap__storageInitializer": "charmedkubeflow/storage-initializer:0.11.2-1bda55b",
-    "serving_runtimes__lgbserver": "kserve/lgbserver:v0.11.1",
+    "serving_runtimes__lgbserver": "charmedkubeflow/lgbserver:0.11.2-1b3817e",
     "serving_runtimes__kserve_mlserver": "docker.io/seldonio/mlserver:1.3.2",
     "serving_runtimes__paddleserver": "charmedkubeflow/paddleserver:0.11.2-ee9909a",
     "serving_runtimes__pmmlserver": "charmedkubeflow/pmmlserver:0.11.2-0a784c4",

diff --git a/charms/kserve-controller/tests/integration/lgbserver.yaml b/charms/kserve-controller/tests/integration/lgbserver.yaml
@@ -0,0 +1,17 @@
+apiVersion: "serving.kserve.io/v1beta1"
+kind: "InferenceService"
+metadata:
+  name: "lightgbm-iris"
+spec:
+  predictor:
+    model:
+      modelFormat:
+        name: lightgbm
+      storageUri: "gs://kfserving-examples/models/lightgbm/iris"
+      resources:
+        limits:
+          cpu: 1
+          memory: 500Mi
+        requests:
+          cpu: 100m
+          memory: 250Mi
diff --git a/charms/kserve-controller/tests/integration/test_charm.py b/charms/kserve-controller/tests/integration/test_charm.py
@@ -243,6 +243,7 @@ def create_namespace():
     "inference_file",
     [
         "./tests/integration/sklearn-iris.yaml",
+        "./tests/integration/lgbserver.yaml",
         "./tests/integration/pmml-server.yaml",
         "./tests/integration/paddleserver-resnet.yaml",
     ],