diff --git a/pyproject.toml b/pyproject.toml index 548a677..50bcdd7 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -54,7 +54,7 @@ dependencies = [ "pytz >=2023.3,<2024.0", "PyYAML >=6.0,<7.0", "requests >=2.25.1, <3.0.0", - "scikit-learn >=1.2.2,<1.3.0", + "scikit-learn >=1.5.0,<1.6.0", "scipy >=1.10.1,<1.11.0", "sentencepiece==0.1.99", "sentence-transformers >=2.2.2,<3.0.0", diff --git a/requirements.txt b/requirements.txt index 7fff657..b4242e1 100644 --- a/requirements.txt +++ b/requirements.txt @@ -219,7 +219,7 @@ ruamel.yaml.clib==0.2.7 ruamel.yaml==0.17.32 s3transfer==0.6.2 safetensors==0.3.1 -scikit-learn==1.2.2 +scikit-learn==1.5.0 scipy==1.10.1 Send2Trash==1.8.2 sentence-transformers==2.2.2