From 91f9f5d4b880ba329924d16b7aa5b86cfcf5cd55 Mon Sep 17 00:00:00 2001
From: Sunyanan Choochotkaew <sunyanan.choochotkaew1@ibm.com>
Date: Mon, 29 Jul 2024 15:31:51 +0900
Subject: [PATCH] apply energy source param on train-model flow

Signed-off-by: Sunyanan Choochotkaew <sunyanan.choochotkaew1@ibm.com>
---
 .github/workflows/train-model.yml | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/.github/workflows/train-model.yml b/.github/workflows/train-model.yml
index 2a08b0f6..3ae0c428 100644
--- a/.github/workflows/train-model.yml
+++ b/.github/workflows/train-model.yml
@@ -166,6 +166,8 @@ jobs:
               persistentVolumeClaim:
                 claimName: task-pvc
             params:
+            - name: ENERGY_SOURCE
+              value: ${{ inputs.energy_source }}
             - name: MODEL_SERVER_IMAGE
               value: ${{ inputs.model_server_image }}
             - name: PIPELINE_NAME