From cdc9e228aaea03f969d8e1aee9833eae2be9272d Mon Sep 17 00:00:00 2001
From: Michael Duong <michael123ste@gmail.com>
Date: Mon, 11 Sep 2023 10:50:07 +0100
Subject: [PATCH] add clean up code for later

---
 .github/workflows/MLPipelinePostMerge.yml | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/.github/workflows/MLPipelinePostMerge.yml b/.github/workflows/MLPipelinePostMerge.yml
index edcf24d..1babf57 100644
--- a/.github/workflows/MLPipelinePostMerge.yml
+++ b/.github/workflows/MLPipelinePostMerge.yml
@@ -44,7 +44,7 @@ jobs:
         dvc push -r dev
 
 
-  Register-New-Model:
+  Register-New-Model-Dev:
     if: github.event.pull_request.merged == true
     runs-on: ubuntu-latest
     steps:
@@ -80,6 +80,7 @@ jobs:
 
   Register-Prediction-Image-Dev:
     needs: Promote-Model-To-Dev
+    # needs: [Promote-Model-To-Dev, Register-New-Model-Dev] WILL ADD BACK ONCE REGISTER WORKS
     runs-on: ubuntu-latest
 
     steps:
@@ -100,10 +101,11 @@ jobs:
         cd modules/ml-pipeline/src/pipeline/src
         dvc pull -r dev
 
-    - name: Build Prediction docker image (TODO - NEED LAMBDA IMAGE)
+    - name: Build Prediction docker image (TODO - NEED LAMBDA IMAGE, need to add version from gto registry)
       run: |
         cd modules/ml-pipeline/src/pipeline/
-        docker build . --file Prediction.Dockerfile --tag prediction_test
+        REGISTER_MODEL_NAME=$(echo ${{ github.event.pull_request.head.ref }} | awk -F"-" '{print $1}')
+        docker build . --file Prediction.Dockerfile --tag ${REGISTER_MODEL_NAME}
 
     - name: ECR Login - Dev
       env: