Add mlflow features

SeldonIO · Sep 14, 2020 · 9077b9c · 9077b9c
1 parent 3b591f1
commit 9077b9c
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/helm-charts/seldon-single-model/templates/seldondeployment.json b/helm-charts/seldon-single-model/templates/seldondeployment.json
@@ -33,6 +33,15 @@
           {{- if .Values.model.uri }}
           "modelUri": "{{ .Values.model.uri }}",
           {{- end}}
+          {{- if .Values.model.mlflow.xtype }}
+          "parameters": [
+		{
+		    "name": "xtype",
+		    "type": "STRING",
+		    "value": "{{ .Values.model.mlflow.xtype }}" 
+		}
+	  ],
+          {{- end}}
           "type": "MODEL",
           {{- if .Values.model.logger.enabled }}
           "logger": {

diff --git a/helm-charts/seldon-single-model/values.yaml b/helm-charts/seldon-single-model/values.yaml
@@ -25,6 +25,8 @@ model:
     enabled: false
     # logger.url -- Remote URL for logger
     url: ""
+  mlflow:
+    xtype: ""
 
 # replicas -- Number of replicas for the predictor
 replicas: 1