diff --git a/requirements.txt b/requirements.txt index d062694..956623a 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,4 +2,4 @@ numpy==1.22.0 gpustat==0.5.0 tqdm==4.32.1 torch==0.4.1 -scikit_learn==0.19.1 +scikit_learn==1.5.0