Fix displaying MLFlow predictors

Signed-off-by: rvalkenaers <[email protected]>
kserve · Sep 15, 2024 · 77b4a01 · 77b4a01
1 parent 704dee3
commit 77b4a01
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/frontend/src/app/shared/utils.ts b/frontend/src/app/shared/utils.ts
@@ -227,6 +227,9 @@ export function getPredictorRuntime(predictor: PredictorSpec): string {
   if (predictorType === PredictorType.Lightgbm) {
     return 'LightGBM ModelServer';
   }
+  if (predictorType === PredictorType.MLFlow) {
+    return 'MLFlow ModelServer';
+  }
   if (predictorType === PredictorType.Custom) {
     return 'Custom ModelServer';
   }

diff --git a/frontend/src/app/types/kfserving/v1beta1.ts b/frontend/src/app/types/kfserving/v1beta1.ts
@@ -47,6 +47,7 @@ export enum PredictorType {
   Xgboost = 'xgboost',
   Pmml = 'pmml',
   Lightgbm = 'lightgbm',
+  MLFlow = 'mlflow',
   Custom = 'custom',
 }
 
@@ -59,6 +60,7 @@ export interface PredictorSpec extends V1PodSpec, ComponentExtensionSpec {
   onnx?: PredictorExtensionSpec;
   pmml?: PredictorExtensionSpec;
   lightgbm?: PredictorExtensionSpec;
+  mlflow?: PredictorExtensionSpec;
   model?: ModelSpec;
 }