Update GitHub CI

ethanluoyc · Nov 6, 2023 · 2c9d48f · 2c9d48f
1 parent 4e2a80e
commit 2c9d48f
Showing 1 changed file with 8 additions and 9 deletions.
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -28,20 +28,19 @@ jobs:
 
     steps:
     - uses: actions/checkout@v3
-
-    - name: Set up Python ${{ matrix.python-version }}
-      uses: actions/setup-python@v4
+    - name: Setup PDM
+      uses: pdm-project/setup-pdm@v3
       with:
         python-version: ${{ matrix.python-version }}
-        cache: 'pip'
+        pdm-version: 2.10.0
+        cache: true
 
-    - name: Install Dependencies
-      run: pip install -e '.[tf,jax,test,dev]'
+    - name: Install dependencies
+      run: pdm install -G tf,jax,test,dev,jax_cuda
 
     - name: Lint
       run: |
-        ruff .
-        black --check --diff .
+        pdm run test
 
     - name: Run tests
-      run: pytest
+      run: pdm run test