improve ci

Author: hiyouga

.github/workflows/tests.yml

@@ -22,7 +22,7 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        python-version:
+        python:
           - "3.9"
           - "3.10"
           - "3.11"
@@ -34,6 +34,10 @@ jobs:
 
     runs-on: ${{ matrix.os }}
 
+    concurrency:
+      group: ${{ github.workflow }}-${{ github.ref }}-${{ matrix.os }}-${{ matrix.python }}
+      cancel-in-progress: ${{ github.ref != 'refs/heads/main' }}
+
     env:
       HF_TOKEN: ${{ secrets.HF_TOKEN }}
       OS_NAME: ${{ matrix.os }}
@@ -45,7 +49,7 @@ jobs:
       - name: Set up Python
         uses: actions/setup-python@v5
         with:
-          python-version: ${{ matrix.python-version }}
+          python-version: ${{ matrix.python }}
           cache: "pip"
           cache-dependency-path: "setup.py"
 
@@ -54,6 +58,12 @@ jobs:
           python -m pip install --upgrade pip
           python -m pip install ".[torch,dev]"
 
+      - name: Cache files
+        uses: actions/cache@v4
+        with:
+          path: ${{ runner.temp }}/huggingface
+          key: huggingface-${{ matrix.os }}-${{ matrix.python }}-${{ hashFiles('tests/version.txt') }}
+
       - name: Check quality
         run: |
           make style && make quality
@@ -64,4 +74,5 @@ jobs:
 
       - name: Test with pytest
         run: |
+          export HF_HOME=${{ runner.temp }}/huggingface
           make test

tests/check_license.py

@@ -20,7 +20,7 @@
 
 
 def main():
-    path_list = []
+    path_list: list[Path] = []
     for check_dir in sys.argv[1:]:
         path_list.extend(Path(check_dir).glob("**/*.py"))
 

tests/version.txt

@@ -0,0 +1 @@
+0.9.3.100