add clean up code for later

This commit is contained in:
Michael Duong 2023-09-11 10:50:07 +01:00
parent 5e0c324938
commit cdc9e228aa

View file

@ -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: