add BASE_IMAGE env variable so the kfp uses the correct base image

redhat-ai-services · Jun 10, 2024 · 5110a54 · 5110a54
1 parent 3f34079
commit 5110a54
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tenants/ai-example/dsp-example-pipeline/base/execute-kfp-task.yaml b/tenants/ai-example/dsp-example-pipeline/base/execute-kfp-task.yaml
@@ -24,5 +24,7 @@ spec:
       env:
         - name: KUBEFLOW_ENDPOINT
           value: $(inputs.params.KUBEFLOW_ENDPOINT)
+        - name: BASE_IMAGE
+          value: $(inputs.params.IMAGE):$(inputs.params.TAG)
       script: |
         python $(inputs.params.SCRIPT)