From 5110a54627bd4e42f246cdb64cc62c1ac4267f93 Mon Sep 17 00:00:00 2001
From: Trevor Royer <troyer@redhat.com>
Date: Mon, 10 Jun 2024 09:18:13 -0600
Subject: [PATCH] add BASE_IMAGE env variable so the kfp uses the correct base
 image

---
 .../ai-example/dsp-example-pipeline/base/execute-kfp-task.yaml  | 2 ++
 1 file changed, 2 insertions(+)

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
index 5a668add..4c343e09 100644
--- 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)