ci: ensure jax is installed for ci tests

rlai-lab · Feb 10, 2025 · 40ed193 · 40ed193
1 parent cceed53
commit 40ed193
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/.github/workflows/main_benchmark.yml b/.github/workflows/main_benchmark.yml
@@ -25,7 +25,7 @@ jobs:
 
       - run: |
           uv venv
-          uv sync
+          uv sync --group=jax
 
       - run: echo "$PWD/.venv/bin" >> $GITHUB_PATH
 

diff --git a/.github/workflows/pr_benchmark.yml b/.github/workflows/pr_benchmark.yml
@@ -20,7 +20,7 @@ jobs:
 
       - run: |
           uv venv
-          uv sync
+          uv sync --group=jax
 
       - run: echo "$PWD/.venv/bin" >> $GITHUB_PATH
 

diff --git a/.github/workflows/pr_test.yml b/.github/workflows/pr_test.yml
@@ -24,7 +24,7 @@ jobs:
 
       - run: |
           uv venv
-          uv sync
+          uv sync --group=jax
 
       - run: echo "$PWD/.venv/bin" >> $GITHUB_PATH
 

diff --git a/.github/workflows/style.yml b/.github/workflows/style.yml
@@ -17,7 +17,7 @@ jobs:
 
       - run: |
           uv venv
-          uv sync
+          uv sync --group=jax
 
       - run: echo "$PWD/.venv/bin" >> $GITHUB_PATH
-Original file line number
+Diff line change
@@ Expand Up / @@ -25,7 +25,7 @@ jobs: @@
           - run: |
               uv venv
-              uv sync
+              uv sync --group=jax
           - run: echo "$PWD/.venv/bin" >> $GITHUB_PATH
@@ Expand Down @@