diff --git a/requirements.txt b/requirements.txt index 0d7d8fc..8b1bacf 100644 --- a/requirements.txt +++ b/requirements.txt @@ -17,7 +17,7 @@ pytorch-metric-learning==2.2.0 requests==2.31.0 retrying==1.3.4 rich==13.3.5 -scikit-learn==1.2.2 +scikit-learn==1.5.0 scipy==1.10.1 seaborn==0.12.2 sentencepiece==0.1.99 diff --git a/setup.py b/setup.py index bccf819..997ee4d 100644 --- a/setup.py +++ b/setup.py @@ -26,7 +26,7 @@ "pytorch-metric-learning==2.2.0", "requests==2.31.0", "rich==13.3.5", - "scikit-learn==1.2.2", + "scikit-learn==1.5.0", "seaborn==0.12.2", "sphinx==5.0.2", "tokenizers==0.15.2",