From c1aba4c5a72d7b0c1cc214f0f2c0b2d47c027d7f Mon Sep 17 00:00:00 2001
From: KevinMusgrave <tkm45@cornell.edu>
Date: Sat, 2 Nov 2024 20:07:04 +0000
Subject: [PATCH] update versions in base_test_workflow

---
 .github/workflows/base_test_workflow.yml | 6 +++---
 setup.py                                 | 2 +-
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/.github/workflows/base_test_workflow.yml b/.github/workflows/base_test_workflow.yml
index 02367536..c697a94c 100644
--- 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
index b8ca0431..66326d46 100644
--- 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",
     ],