From 31d77bdb2e8de59e3ade40be38d4761067490a21 Mon Sep 17 00:00:00 2001 From: Hansini Karunarathne <107214435+hansinikarunarathne@users.noreply.github.com> Date: Wed, 4 Sep 2024 20:07:41 +0530 Subject: [PATCH] Revert "Allow custom ServingRuntimes (#92)" (#93) * Revert "Allow custom ServingRuntimes (#92)" This reverts commit 4675bb95805b7197c340b762ec07a5a0fd06d7cd. Signed-off-by: hansinikarunarathne * Add prettier to makefile Signed-off-by: hansinikarunarathne --------- Signed-off-by: hansinikarunarathne --- Makefile | 2 +- frontend/src/app/shared/utils.ts | 9 ++++++++- 2 files changed, 9 insertions(+), 2 deletions(-) diff --git a/Makefile b/Makefile index 90ce0a9..7f47bf2 100644 --- a/Makefile +++ b/Makefile @@ -2,11 +2,11 @@ IMG ?= kserve/models-web-app TAG ?= $(shell git describe --tags --always --dirty) ARCH ?= linux/amd64 + # Prettier UI format check. prettier-check: npm run format:check --prefix frontend - docker-build: docker build -t ${IMG}:${TAG} . diff --git a/frontend/src/app/shared/utils.ts b/frontend/src/app/shared/utils.ts index 6b78ce8..f637e92 100644 --- a/frontend/src/app/shared/utils.ts +++ b/frontend/src/app/shared/utils.ts @@ -143,7 +143,14 @@ export function getPredictorExtensionSpec( predictor: PredictorSpec, ): PredictorExtensionSpec { if (predictor.model) { - return predictor.model; + if ( + Object.values(PredictorType).includes( + predictor.model?.modelFormat.name as PredictorType, + ) + ) { + const spec = predictor.model; + return spec; + } } else { for (const predictorType of Object.values(PredictorType)) { if (predictorType in predictor) {