diff --git a/.github/workflows/MLPipelinePostMerge.yml b/.github/workflows/MLPipelinePostMerge.yml new file mode 100644 index 0000000..660acd0 --- /dev/null +++ b/.github/workflows/MLPipelinePostMerge.yml @@ -0,0 +1,42 @@ +name: Register the model for the given pipeline branch + +# on: +# push: +# branches: +# - "model-**" + +on: + pull_request: + types: + - closed + branches: + - "master" + +permissions: write-all + +jobs: + Register-New-Model: + if: github.event.pull_request.merged == true + runs-on: ubuntu-latest + steps: + - uses: actions/checkout@v3 + with: + fetch-depth: 0 + + - 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: | + REGISTER_MODEL_NAME=$(echo ${{github.ref_name}} | awk -F"-" '{print $1}') + git config user.name "Github-Bot" + git config user.email "Github-Bot" + + gto register ${REGISTER_MODEL_NAME} + gto assign ${REGISTER_MODEL_NAME} --stage dev + gto show diff --git a/.github/workflows/MLPipelinePullRequest.yml b/.github/workflows/MLPipelinePullRequest.yml index 186c382..614a984 100644 --- a/.github/workflows/MLPipelinePullRequest.yml +++ b/.github/workflows/MLPipelinePullRequest.yml @@ -1,9 +1,9 @@ name: Build and test model using a dummy prediction pipeline on: - push: - branches: - - "model-**" + # push: + # branches: + # - "model-**" pull_request: branches: [ "master" ] @@ -74,7 +74,7 @@ jobs: # Compare metrics to master git fetch --depth=1 origin master:master - dvc metrics diff --md master >> report.md + dvc metrics diff --md --all master >> report.md cml comment create report.md