Revert "Allow custom ServingRuntimes (#92)" (#93)

* Revert "Allow custom ServingRuntimes (#92)" This reverts commit 4675bb9. Signed-off-by: hansinikarunarathne <[email protected]> * Add prettier to makefile Signed-off-by: hansinikarunarathne <[email protected]> --------- Signed-off-by: hansinikarunarathne <[email protected]>
kserve · Sep 4, 2024 · 31d77bd · 31d77bd
1 parent 4675bb9
commit 31d77bd
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git 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
@@ -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) {