migrate environment installs to uv

https://github.com/astral-sh/uv
kornia · Mar 11, 2024 · f8e904c · f8e904c
1 parent 3ae3c00
commit f8e904c
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 6 deletions.
diff --git a/.github/actions/env/action.yml b/.github/actions/env/action.yml
@@ -39,24 +39,28 @@ runs:
       with:
         python-version: ${{ inputs.python-version }}
 
+    - name: Install UV
+      shell: bash
+      run: pip install uv
+
     - if: ${{ contains(fromJson('["nightly"]'), inputs.pytorch-version ) }}
       name: Install PyTorch nightly
       shell: bash
-      run: pip install numpy --pre torch[dynamo] ${{ inputs.extra-deps }} --force-reinstall --extra-index-url https://download.pytorch.org/whl/nightly/cpu/
+      run: uv pip install --system numpy --pre torch[dynamo] ${{ inputs.extra-deps }} --force-reinstall --extra-index-url https://download.pytorch.org/whl/nightly/cpu/
 
     - if: ${{ contains(fromJson('["nightly"]'), inputs.pytorch-version ) == false}}
       name: Install pytorch
       shell: bash
-      run: pip install torch==${{ inputs.pytorch-version }} ${{ inputs.extra-deps }} --index-url https://download.pytorch.org/whl/cpu
+      run: uv pip install --system torch==${{ inputs.pytorch-version }} ${{ inputs.extra-deps }} --index-url https://download.pytorch.org/whl/cpu
 
     - if: ${{ contains(fromJson('["1.9.1"]'), inputs.pytorch-version) }}
       name: Install accelerate for old torchs
       shell: bash
-      run: pip install accelerate==0.20.3
+      run: uv pip install --system accelerate==0.20.3
 
     - name: Install Kornia dev
       shell: bash
-      run: pip install .[dev,x]
+      run: uv pip install --system --editable .[dev,x]
 
     - name: Check torch version
       shell: bash

diff --git a/.github/workflows/coverage.yml b/.github/workflows/coverage.yml
@@ -101,9 +101,13 @@ jobs:
         with:
           python-version: ${{ inputs.python-version }}
 
+      - name: Install UV
+        shell: bash
+        run: pip install uv
+
       - name: Install dev dependencies
         shell: bash
-        run: pip install -r ./requirements/requirements-dev.txt
+        run: uv pip install --system -r ./requirements/requirements-dev.txt
 
       - name: Run coverage
         run: |

diff --git a/.github/workflows/docs.yml b/.github/workflows/docs.yml
@@ -61,7 +61,7 @@ jobs:
 
     - name: Install docs deps
       shell: bash -l {0}
-      run: pip install -e .[docs]
+      run: uv pip install --system --editable .[docs]
 
     - name: Build Documentation
       shell: bash -l {0}