From 17f96971086bd62f05cd4d18a544ebdc9883bd14 Mon Sep 17 00:00:00 2001
From: quandanrepo <45804868+quandanrepo@users.noreply.github.com>
Date: Sat, 16 Sep 2023 12:00:00 +0100
Subject: [PATCH] Update MLPipelinePostMerge.yml

---
 .github/workflows/MLPipelinePostMerge.yml | 65 +++++++++++------------
 1 file changed, 32 insertions(+), 33 deletions(-)

diff --git a/.github/workflows/MLPipelinePostMerge.yml b/.github/workflows/MLPipelinePostMerge.yml
index 1767cd6..b8ab439 100644
--- a/.github/workflows/MLPipelinePostMerge.yml
+++ b/.github/workflows/MLPipelinePostMerge.yml
@@ -48,7 +48,7 @@ jobs:
         git tag -a ${new_tag} -m "Registering new Major Version"
         git push origin ${new_tag}
 
-        gto show >> MODEL_REGISTRY.md
+        gto show > MODEL_REGISTRY.md
         git add .
         git commit -m "Update Registry"
         git push
@@ -86,7 +86,7 @@ jobs:
         git tag -a ${new_tag} -m "Registering new Minor Version"
         git push origin ${new_tag}
 
-        gto show >> MODEL_REGISTRY.md
+        gto show > MODEL_REGISTRY.md
         git add .
         git commit -m "Update Registry"
         git push
@@ -124,7 +124,7 @@ jobs:
         git tag -a ${new_tag} -m "Registering new Patch Version"
         git push origin ${new_tag}
 
-        gto show >> MODEL_REGISTRY.md
+        gto show > MODEL_REGISTRY.md
         git add .
         git commit -m "Update Registry"
         git push
@@ -156,44 +156,43 @@ jobs:
         cd modules/ml-pipeline/src/pipeline/src
         dvc push -r dev
 
+  Register-New-Model-Dev:
+    if: github.event.pull_request.merged == true
+    runs-on: ubuntu-latest
+    steps:
+    - uses: actions/checkout@v3
+      with:
+        fetch-depth: 0
 
-  # Register-New-Model-Dev:
-  #   if: github.event.pull_request.merged == true
-  #   runs-on: ubuntu-latest
-  #   steps:
-  #   - uses: actions/checkout@v4
-  #     with:
-  #       fetch-depth: 0
+    - name: Install packages to register model
+      run: |
+        pip install --upgrade pip
+        pip install -r modules/ml-pipeline/src/pipeline/src/requirements/version_control/requirements.txt
 
-  #   - name: Install packages to register model
-  #     env:
-  #       AWS_ACCESS_KEY_ID: ${{ secrets.ROBOT_AWS_ACCESS_KEY_ID }}
-  #       AWS_SECRET_ACCESS_KEY: ${{ secrets.ROBOT_AWS_SECRET_ACCESS_KEY }}
-  #     run: |
-  #       pip install --upgrade pip
-  #       pip install -r modules/ml-pipeline/src/pipeline/src/requirements/version_control/requirements.txt
+    - name: Register Model
+      env:
+        AWS_ACCESS_KEY_ID: ${{ secrets.ROBOT_AWS_ACCESS_KEY_ID }}
+        AWS_SECRET_ACCESS_KEY: ${{ secrets.ROBOT_AWS_SECRET_ACCESS_KEY }}
+      run: |
 
-  #   - name: Register Model
-  #     env:
-  #       AWS_ACCESS_KEY_ID: ${{ secrets.ROBOT_AWS_ACCESS_KEY_ID }}
-  #       AWS_SECRET_ACCESS_KEY: ${{ secrets.ROBOT_AWS_SECRET_ACCESS_KEY }}
-  #     run: |
+        REGISTER_MODEL_NAME=$(echo ${{ github.event.pull_request.head.ref }} | awk -F"-" '{print $1}')
+        # REGISTER_MODEL_NAME=$(echo ${{github.ref_name}} | awk -F"-" '{print $1}')
 
-  #       # REGISTER_MODEL_NAME=$(echo ${{ github.event.pull_request.head.ref }} | awk -F"-" '{print $1}')
-  #       REGISTER_MODEL_NAME=$(echo ${{github.ref_name}} | awk -F"-" '{print $1}')
+        git config user.name "Github-Bot"
+        git config user.email "Github-Bot@no-reply.com"
 
-  #       git config user.name "Github-Bot"
-  #       git config user.email "Github-Bot@no-reply.com"
+        latest_version=$(gto show ${REGISTER_MODEL_NAME}@latest --ref)
+        
+        new_tag=${latest_version}#dev
 
-  #       # git tag model@v0.0.1
-  #       # git push origin model@v0.0.1
+        git tag -a ${new_tag} -m "Registering Latest Version to Dev"
+        git push origin ${new_tag}
 
-  #       # gto register test --repo https://github.com/Hestia-Homes/ML/
-  #       # echo "chicken" >> test.md
+        gto show > MODEL_REGISTRY.md
+        git add .
+        git commit -m "Update Registry"
+        git push
 
-  #       # gto -v register ${REGISTER_MODEL_NAME}
-  #       # gto assign regression --stage dev
-  #       # gto show
 
   Register-Prediction-Image-Dev:
     needs: Promote-Artefacts-To-Dev