From 2c9d48fb5a901bcb2b35b6f391aca0d3416af920 Mon Sep 17 00:00:00 2001
From: Yicheng Luo <ethanluoyc@gmail.com>
Date: Mon, 6 Nov 2023 22:46:55 +0000
Subject: [PATCH] Update GitHub CI

---
 .github/workflows/test.yml | 17 ++++++++---------
 1 file changed, 8 insertions(+), 9 deletions(-)

diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index 2f3abbe..1b11fe1 100644
--- 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