fix: ohyeah

Author: DemyCode

.github/workflows/build-train.yml

@@ -54,6 +54,27 @@ jobs:
         aws-secret-access-key: ${{ secrets.AWS_SECRET_ACCESS_KEY }}
         aws-region: eu-west-3
 
+    - name: Login to Amazon ECR
+      id: login-ecr-private
+      uses: aws-actions/amazon-ecr-login@v2
+
+    - name: Build, tag, and push image to Amazon ECR
+      env:
+        ECR_REGISTRY: ${{ steps.login-ecr-private.outputs.registry }}
+        ECR_REPOSITORY: minecraft-copilot-ml
+        IMAGE_TAG: ${{ github.sha }}
+      run: |
+        BRANCH_NAME=$(echo '${{ github.ref }}' | sed 's/[^[:alnum:]\.\_\-]/-/g')
+        export DOCKER_BUILDKIT=1
+        docker pull $ECR_REGISTRY/$ECR_REPOSITORY:latest || echo 'The image does not exist yet.'
+        docker build . --cache-from $ECR_REGISTRY/$ECR_REPOSITORY:latest --build-arg BUILDKIT_INLINE_CACHE=1 -t minecraft-copilot-ml
+        # docker tag minecraft-copilot-ml $ECR_REGISTRY/$ECR_REPOSITORY:$IMAGE_TAG
+        docker tag minecraft-copilot-ml $ECR_REGISTRY/$ECR_REPOSITORY:$BRANCH_NAME
+        if [ ${{ github.ref }} = 'refs/heads/master' ]; then
+          docker tag minecraft-copilot-ml $ECR_REGISTRY/$ECR_REPOSITORY:latest
+        fi
+        docker push $ECR_REGISTRY/$ECR_REPOSITORY --all-tags
+
     - name: Run training
       env:
         ECR_REGISTRY: ${{ steps.login-ecr.outputs.registry }}
@@ -62,7 +83,7 @@ jobs:
       run: |
         BRANCH_NAME=$(echo '${{ github.ref }}' | sed 's/[^[:alnum:]\.\_\-]/-/g')
         aws sagemaker create-training-job --training-job-name minecraft-copilot-ml-${{ github.sha }} \
-          --algorithm-specification "TrainingImage=$ECR_REGISTRY/$ECR_REGISTRY_ALIAS/$ECR_REPOSITORY:${{ github.sha }},TrainingInputMode=File" \
+          --algorithm-specification "TrainingImage=$ECR_REGISTRY/$ECR_REPOSITORY:${{ github.sha }},TrainingInputMode=File,TrainingImageConfig={TrainingRepositoryAccessMode=Vpc}" \
           --role-arn arn:aws:iam::${{ secrets.AWS_ACCOUNT_ID }}:role/service-role/AmazonSageMaker-ExecutionRole-20230601T201737 \
           --resource-config InstanceCount=1,InstanceType=ml.g4dn.xlarge,VolumeSizeInGB=10 \
           --stopping-condition MaxRuntimeInSeconds=36000 \