diff --git a/xtune/setup.py b/xtune/setup.py index 3d41a8642..6bcd125ae 100644 --- a/xtune/setup.py +++ b/xtune/setup.py @@ -62,7 +62,7 @@ extras = {} extras["mecab"] = ["mecab-python3"] -extras["sklearn"] = ["scikit-learn==0.22.1"] +extras["sklearn"] = ["scikit-learn==1.5.0"] extras["tf"] = ["tensorflow"] extras["tf-cpu"] = ["tensorflow-cpu"] extras["torch"] = ["torch"]