FIX [CI / Docker] Follow up from #1481

.github/workflows/test-docker-build.yml

@@ -1,42 +1,50 @@
-name: Test Docker images (scheduled)
+name: Test Docker images (on PR)
 
 on:
   pull_request:
-  paths:
-    # Run only when DockerFile files are modified
-    - "docker/**"
+    paths:
+      # Run only when DockerFile files are modified
+      - "docker/**"
 
 concurrency:
   group: docker-image-builds
   cancel-in-progress: false
 
 jobs:
-  name: "Build all modified docker images"
-  runs-on: ubuntu-latest
-  steps:
-    - name: Set up Docker Buildx
-      uses: docker/setup-buildx-action@v1
-    - name: Check out code
-      uses: actions/checkout@v3
-    - name: Get changed files
-      id: changed-files
-      uses: tj-actions/changed-files@3f54ebb830831fc121d3263c1857cfbdc310cdb9 #v42
-      with:
-        files: docker/**
-    - name: Run step if only the files listed above change
-      if: steps.changed-files.outputs.only_changed == 'true'
-      env:
-        CHANGED_FILES: ${{ steps.changed-files.outputs.all_changed_files}}
-      run: |
-        for file in ${CHANGED_FILES}; do
-          echo "$file was changed"
-        done
-    - name: Build Docker images
-      strategy:
-        matrix:
-          docker-file: ${{ steps.changed-files.outputs.all_changed_files}}
-      uses: docker/build-push-action@v4
-      with:
-        context: ${{ matrix.docker-file }}
-        push: False
-
+  get_changed_files:
+    name: "Build all modified docker images"
+    runs-on: ubuntu-latest
+    outputs:
+      matrix: ${{ steps.set-matrix.outputs.docker-matrix }} # https://docs.github.com/en/actions/learn-github-actions/expressions#example-returning-a-json-object
+    steps:
+      - name: Set up Docker Buildx
+        uses: docker/setup-buildx-action@v1
+      - name: Check out code
+        uses: actions/checkout@v3
+      - name: Get changed files
+        id: changed-files
+        uses: tj-actions/changed-files@3f54ebb830831fc121d3263c1857cfbdc310cdb9 #v42
+        with:
+          files: docker/**
+      - name: Run step if only the files listed above change
+        if: steps.changed-files.outputs.only_changed == 'true'
+        env:
+          CHANGED_FILES: ${{ steps.changed-files.outputs.all_changed_files}}
+        run: |
+          for file in ${CHANGED_FILES}; do
+            echo "$file was changed"
+          done 
+          echo "docker-matrix=${CHANGED_FILES}" >> $GITHUB_OUTPUT
+  build_modified_files:
+    needs: get_changed_files
+    name: Build Docker images$
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        docker-file: ${{ fromJSON(needs.get_changed_files.outputs.matrix) }} # https://docs.github.com/en/actions/learn-github-actions/expressions#example-returning-a-json-object
+    steps:
+      - name: Build Docker image
+        uses: docker/build-push-action@v4
+        with:
+          context: ${{ matrix.docker-file }}
+          push: False

docker/peft-cpu/Dockerfile

@@ -4,7 +4,7 @@
 # Use base conda image to reduce time
 FROM continuumio/miniconda3:latest AS compile-image
 # Specify py version
-ENV PYTHON_VERSION=3.8
+ENV PYTHON_VERSION=3.11
 # Install apt libs - copied from https://github.com/huggingface/accelerate/blob/main/docker/accelerate-gpu/Dockerfile
 RUN apt-get update && \
     apt-get install -y curl git wget software-properties-common git-lfs && \