From e7b77aa7a2f31e184dc5489d7134690a149279cb Mon Sep 17 00:00:00 2001
From: Shuli Shu <08cnbj@gmail.com>
Date: Tue, 7 Jan 2025 18:08:12 +0000
Subject: [PATCH] update

---
 .../workflows/wheel_linux_aarch64_cuda.yml    | 27 +++++++++++--------
 1 file changed, 16 insertions(+), 11 deletions(-)

diff --git a/.github/workflows/wheel_linux_aarch64_cuda.yml b/.github/workflows/wheel_linux_aarch64_cuda.yml
index 789124a70..612c03e18 100644
--- 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,6 +113,7 @@ jobs:
             python -m pip install $wheel
           done
           cd ..
+
           DEVICENAME=(echo ${{ matrix.pl_backend }} | %{$_ -replace "_","."})
           if ${{ matrix.pl_backend == 'lightning_gpu'}}
           then
@@ -115,11 +121,10 @@ jobs:
           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