From faddfbf7cad712b593ab4808a1dc0f761348a9af Mon Sep 17 00:00:00 2001
From: Michael Duong <michael123ste@gmail.com>
Date: Sun, 10 Sep 2023 20:45:51 +0100
Subject: [PATCH] test gto step

---
 .github/workflows/MLPipelinePostMerge.yml | 29 +++++++++++------------
 1 file changed, 14 insertions(+), 15 deletions(-)

diff --git a/.github/workflows/MLPipelinePostMerge.yml b/.github/workflows/MLPipelinePostMerge.yml
index 3d4d27b..ded8c15 100644
--- a/.github/workflows/MLPipelinePostMerge.yml
+++ b/.github/workflows/MLPipelinePostMerge.yml
@@ -19,19 +19,18 @@ jobs:
     # if: github.event.pull_request.merged == true
     runs-on: ubuntu-latest
     steps:
-    # - uses: actions/checkout@v3
-    # - 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
-    #   run: |
-
-    #     dvc pull -r experiments
-    - name: echo branch
+    - uses: actions/checkout@v3
+    - 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: |
-        register_name=$(echo ${{github.ref_name}} | awk -F"-" '{print $1}')
+        pip install --upgrade pip
+        pip install -r modules/ml-pipeline/src/pipeline/src/requirements/version_control/requirements.txt
+
+    - name: Register Model
+      run: |
+        REGISTER_MODEL_NAME=$(echo ${{github.ref_name}} | awk -F"-" '{print $1}')
+        gto register ${REGISTER_MODEL_NAME}
+        gto assign ${REGISTER_MODEL_NAME} --stage dev
+        gto show