diff --git a/requirements.txt b/requirements.txt index d6be752..2a0ed03 100644 --- a/requirements.txt +++ b/requirements.txt @@ -3,4 +3,4 @@ numpy joblib==1.3.1 torch==2.0.1 tqdm==4.66.3 -scikit-learn==1.3.0 +scikit-learn==1.5.0 diff --git a/setup.py b/setup.py index bf06f11..90f8800 100644 --- a/setup.py +++ b/setup.py @@ -105,7 +105,7 @@ def run_tests(self): "torch==2.0.1", "joblib==1.3.1", "tqdm==4.66.3", - "scikit-learn==1.3.0", + "scikit-learn==1.5.0", ], # List additional groups of dependencies here (e.g. development # dependencies). You can install these using the following syntax,