update

PennyLaneAI · Jan 7, 2025 · e7b77aa · e7b77aa
1 parent f3d4167
commit e7b77aa
Showing 1 changed file with 16 additions and 11 deletions.
diff --git a/.github/workflows/wheel_linux_aarch64_cuda.yml b/.github/workflows/wheel_linux_aarch64_cuda.yml
@@ -47,17 +47,11 @@ jobs:
       - ${{ matrix.os }}
 
     steps:
-
       - name: Checkout PennyLane-Lightning
         uses: actions/checkout@v4
 
-      - name: Install Python 3.10
-        uses: actions/setup-python@v5
-        with:
-          python-version: '3.10'
-
       - name: Install cibuildwheel
-        run: python3.10 -m pip install cibuildwheel~=2.20.0 tomlkit
+        run: python -m pip install cibuildwheel~=2.20.0 tomlkit
 
       - name: Configure pyproject.toml file
         run: PL_BACKEND="${{ matrix.pl_backend }}" python scripts/configure_pyproject_toml.py
@@ -92,7 +86,18 @@ jobs:
 
           CIBW_BUILD_VERBOSITY: 1
 
-        run: python3 -m cibuildwheel --output-dir wheelhouse
+        run: python -m cibuildwheel --output-dir wheelhouse
+
+      - name: Determine Python version
+        id: pyvs
+        shell: bash
+        run: |
+          echo "version=$(echo  ${{ matrix.cibw_build }} | tr -cd '[:digit:].' | sed 's/./&./1')" >> $GITHUB_OUTPUT
+
+      - uses: actions/setup-python@v5
+        name: Install Python
+        with:
+          python-version: ${{ steps.pyvs.outputs.version }}
 
       - name: Test wheels
         run: |
@@ -108,18 +113,18 @@ jobs:
             python -m pip install $wheel
           done
           cd ..
+
           DEVICENAME=(echo ${{ matrix.pl_backend }} | %{$_ -replace "_","."})
           if ${{ matrix.pl_backend == 'lightning_gpu'}}
           then
             pl-device-test --device=$DEVICENAME --skip-ops -x --tb=short --no-flaky-report
           fi
           PL_DEVICE=${DEVICENAME} python -m pytest tests/ -k "not test_native_mcm"
 
-
       - name: Validate wheels
         run: |
-         python3 -m pip install twine
-         python3 -m twine check ./wheelhouse/*.whl
+         python -m pip install twine
+         python -m twine check ./wheelhouse/*.whl
 
       - uses: actions-ecosystem/action-regex-match@main
         id: rc_build