update versions in base_test_workflow

KevinMusgrave · Nov 2, 2024 · c1aba4c · c1aba4c
1 parent 3b37166
commit c1aba4c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/.github/workflows/base_test_workflow.yml b/.github/workflows/base_test_workflow.yml
@@ -18,8 +18,8 @@ jobs:
             pytorch-version: 1.6
             torchvision-version: 0.7
           - python-version: 3.9
-            pytorch-version: 2.3
-            torchvision-version: 0.18
+            pytorch-version: 2.5
+            torchvision-version: 0.20
 
     steps:
     - uses: actions/checkout@v2
@@ -30,7 +30,7 @@ jobs:
     - name: Install dependencies
       run: |
         pip install .[with-hooks-cpu]
-        pip install "numpy<2.0" torch==${{ matrix.pytorch-version }} torchvision==${{ matrix.torchvision-version }} --force-reinstall
+        pip install torch==${{ matrix.pytorch-version }} torchvision==${{ matrix.torchvision-version }} --force-reinstall
         pip install --upgrade protobuf==3.20.1
         pip install six
         pip install packaging

diff --git a/setup.py b/setup.py
@@ -40,7 +40,7 @@
     python_requires=">=3.0",
     install_requires=[
         "numpy",
-        "scikit-learn <= 1.4.1.post1",
+        "scikit-learn",
         "tqdm",
         "torch >= 1.6.0",
     ],