github actions

lucidrains · Oct 29, 2024 · 7f4e6ce · 7f4e6ce
1 parent c9d46a4
commit 7f4e6ce
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/.github/workflows/python-publish.yml b/.github/workflows/python-publish.yml
@@ -25,7 +25,7 @@ jobs:
         python-version: '3.x'
     - name: Install dependencies
       run: |
-        python -m pip install --upgrade pip
+        python -m pip install --upgrade pip uv
         pip install build
     - name: Build package
       run: python -m build

diff --git a/.github/workflows/python-test.yaml b/.github/workflows/python-test.yaml
@@ -25,8 +25,8 @@ jobs:
         python-version: ${{ matrix.python-version }}
     - name: Install dependencies
       run: |
-        python -m pip install --upgrade pip
-        python -m pip install torch==2.1.0 --index-url https://download.pytorch.org/whl/cpu
+        python -m pip install --upgrade pip uv
+        python -m uv pip install torch==2.1.0 --index-url https://download.pytorch.org/whl/cpu
     - name: Test with pytest
       run: |
         python setup.py test