Merge pull request #79 from tasos-ale/feature-tasos-custom-predictor

mwa(front): Fix custom predictor bug
kserve · May 22, 2024 · 880e635 · 880e635
2 parents 775b93b + 3c74656
commit 880e635
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/frontend/src/app/shared/utils.ts b/frontend/src/app/shared/utils.ts
@@ -161,12 +161,12 @@ export function getPredictorExtensionSpec(
 
   // In the case of Custom predictors, set the additional PredictorExtensionSpec fields
   // manually here
-  const spec = predictor.containers[0] as PredictorExtensionSpec;
+  const spec = (predictor?.containers?.[0] || {}) as PredictorExtensionSpec;
   spec.runtimeVersion = '';
   spec.protocolVersion = '';
 
-  if (predictor.containers[0]?.env) {
-    const storageUri = predictor.containers[0].env.find(
+  if (spec?.env) {
+    const storageUri = spec.env.find(
       envVar => envVar.name.toLowerCase() === 'storage_uri',
     );
     if (storageUri) {