Use uv to download packages on github CI

Stable-Baselines-Team · Oct 24, 2024 · f1fc8f5 · f1fc8f5
1 parent 72abe85
commit f1fc8f5
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -30,21 +30,23 @@ jobs:
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip
+          # Use uv for faster downloads
+          pip install uv
           # cpu version of pytorch
-          pip install torch==2.1.1 --index-url https://download.pytorch.org/whl/cpu
+          uv pip install torch==2.1.1 --index-url https://download.pytorch.org/whl/cpu
 
           # Install Atari Roms
-          pip install autorom
+          uv pip install autorom
           wget https://gist.githubusercontent.com/jjshoots/61b22aefce4456920ba99f2c36906eda/raw/00046ac3403768bfe45857610a3d333b8e35e026/Roms.tar.gz.b64
           base64 Roms.tar.gz.b64 --decode &> Roms.tar.gz
           AutoROM --accept-license --source-file Roms.tar.gz
 
           # Install master version
           # and dependencies for docs and tests
-          pip install "stable_baselines3[extra_no_roms,tests,docs] @ git+https://github.com/DLR-RM/stable-baselines3"
-          pip install .
+          uv pip install "stable_baselines3[extra_no_roms,tests,docs] @ git+https://github.com/DLR-RM/stable-baselines3"
+          uv pip install .
           # Use headless version
-          pip install opencv-python-headless
+          uv pip install opencv-python-headless
 
       - name: Lint with ruff
         run: |

diff --git a/docs/misc/changelog.rst b/docs/misc/changelog.rst
@@ -31,6 +31,7 @@ Others:
 ^^^^^^^
 - Updated PyTorch version on CI to 2.3.1
 - Remove unnecessary SDE noise resampling in PPO/TRPO update
+- Switched to uv to download packages on GitHub CI
 
 Documentation:
 ^^^^^^^^^^^^^^