github actions (lucidrains#285)

fix
iamthebot · Oct 29, 2024 · 6e56f43 · 6e56f43
1 parent c9d46a4
commit 6e56f43
Show file tree

Hide file tree

Showing 2 changed files with 4 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,9 @@ 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+cpu --extra-index-url https://download.pytorch.org/whl/cpu
+        python -m uv pip install .
     - name: Test with pytest
       run: |
         python setup.py test