Try uv

NVIDIA · Jan 13, 2025 · 7153f43 · 7153f43
1 parent 532368f
commit 7153f43
Show file tree

Hide file tree

Showing 4 changed files with 25 additions and 19 deletions.
diff --git a/.github/workflows/doc_build.yml b/.github/workflows/doc_build.yml
@@ -17,11 +17,12 @@ jobs:
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip
-          python -m pip install pytest
-          python -m pip install ./cuequivariance
-          python -m pip install ./cuequivariance_jax
-          python -m pip install ./cuequivariance_torch
-          pip install -r docs/requirements.txt
+          python -m pip install --upgrade uv
+          python -m uv pip install pytest
+          python -m uv pip install ./cuequivariance
+          python -m uv pip install ./cuequivariance_jax
+          python -m uv pip install ./cuequivariance_torch
+          python -m uv pip install -r docs/requirements.txt
       - name: Build sphinx
         run: |
           sphinx-build -b html docs docs/public
diff --git a/.github/workflows/doc_build_deploy.yml b/.github/workflows/doc_build_deploy.yml
@@ -38,11 +38,12 @@ jobs:
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip
-          python -m pip install pytest
-          python -m pip install ./cuequivariance
-          python -m pip install ./cuequivariance_jax
-          python -m pip install ./cuequivariance_torch
-          pip install -r docs/requirements.txt
+          python -m pip install --upgrade uv
+          python -m uv pip install pytest
+          python -m uv pip install ./cuequivariance
+          python -m uv pip install ./cuequivariance_jax
+          python -m uv pip install ./cuequivariance_torch
+          python -m uv pip install -r docs/requirements.txt
       - name: Build sphinx
         run: |
           sphinx-build -b html docs docs/public

diff --git a/.github/workflows/style.yml b/.github/workflows/style.yml
@@ -23,7 +23,8 @@ jobs:
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
-        python -m pip install ruff==0.8.2
+        python -m pip install --upgrade uv
+        python -m uv pip install ruff==0.8.2
     - name: Style check
       run: |
         ruff check cuequivariance

diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -25,8 +25,9 @@ jobs:
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
-        python -m pip install pytest
-        python -m pip install ./cuequivariance
+        python -m pip install --upgrade uv
+        python -m uv pip install pytest
+        python -m uv pip install ./cuequivariance
     - name: Test with pytest
       run: |
         pytest --doctest-modules -x cuequivariance
@@ -48,9 +49,10 @@ jobs:
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
-        python -m pip install pytest
-        python -m pip install ./cuequivariance
-        python -m pip install ./cuequivariance_jax
+        python -m pip install --upgrade uv
+        python -m uv pip install pytest
+        python -m uv pip install ./cuequivariance
+        python -m uv pip install ./cuequivariance_jax
     - name: Test with pytest
       run: |
         pytest --doctest-modules -x cuequivariance_jax
@@ -78,9 +80,10 @@ jobs:
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
-        python -m pip install pytest torch e3nn
-        python -m pip install ./cuequivariance
-        python -m pip install ./cuequivariance_torch
+        python -m pip install --upgrade uv
+        python -m uv pip install pytest torch e3nn
+        python -m uv pip install ./cuequivariance
+        python -m uv pip install ./cuequivariance_torch
     - name: Test with pytest
       run: |
         pytest --doctest-modules -x cuequivariance_torch