ci: use uv in place of pip to speed up installing dependencies

wookayin · Jan 23, 2025 · 8aa9ea9 · 8aa9ea9
1 parent 116d087
commit 8aa9ea9
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 7 deletions.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -40,17 +40,16 @@ jobs:
         with:
           python-version: ${{ matrix.python-version }}
 
-      - name: Upgrade pip
-        run: |
-          python -m pip install -U pip
+      - name: Install uv
+        uses: astral-sh/setup-uv@v5
 
       - name: Configure environments
         run: |
           python --version
 
       - name: Install dependencies
         run: |
-          pip install -e ".[test]"
+          uv pip install -e ".[test]"
           python -m imgcat --version
 
       - name: Run tests

diff --git a/setup.py b/setup.py
@@ -28,9 +28,9 @@ def read_version():
 
 tests_requires = [
     'pytest',
-    'numpy',
-    'torch',
-    'tensorflow>=2.0',
+    'numpy>=1.21,<2.0',
+    'torch>=2.0,<3.0',
+    'tensorflow>=2.0,<3.0',
     'matplotlib>=3.3',
     'Pillow',
 ]