[CI] auto clean cache & retry

Author: CSY-ModelCloud

.github/workflows/unit_tests.yml

@@ -286,6 +286,36 @@ jobs:
           
           ${{ matrix.install_cmd }}
 
+      - name: Clear installer cache on failure
+        if: failure()
+        run: |
+          INSTALL_CMD='${{ matrix.install_cmd }}'
+
+          if [[ "$INSTALL_CMD" == uv* || "$INSTALL_CMD" == *" uv "* ]]; then
+            echo "[INFO] Previous install command uses uv, clearing uv cache..."
+            uv cache clean
+          elif [[ "$INSTALL_CMD" == pip* || "$INSTALL_CMD" == *" pip "* ]]; then
+            echo "[INFO] Previous install command uses pip, clearing pip cache..."
+            python -m pip cache purge || rm -rf ~/.cache/pip
+          fi
+          echo "RETRY_INSTALL=true" >> $GITHUB_ENV
+
+      - name: retry ${{ matrix.name }}
+        if: env.RETRY_INSTALL == 'true'
+        run: |
+          uv venv "${{ matrix.env_name }}"
+          source "${{ matrix.env_name }}/bin/activate"
+          if [[ "${{ matrix.use_pip }}" == "true" ]]; then
+            uv pip install pip -U
+          fi
+          if python -VV 2>&1 | grep -q '3\.14.*free-threading'; then
+            echo "[INFO] Python 3.14t detected, installing torchao..."
+            uv pip install http://10.0.13.31/files/torchao-0.18.0+git13cd013d6-cp314-cp314t-linux_x86_64.whl
+          else
+            echo "[INFO] Current Python is not 3.14t, skip torchao."
+          fi
+          
+          ${{ matrix.install_cmd }}
 
   setuptools-compatibility:
     uses: ./.github/workflows/setuptools_compatibility_reusable.yml