diff --git a/setup.py b/setup.py index dd9f969..996a11b 100644 --- a/setup.py +++ b/setup.py @@ -8,7 +8,7 @@ 'sdv', 'torch<2,>=1.0', 'torchvision<1,>=0.4.2', - 'scikit-learn==0.24.1', + 'scikit-learn==1.0.1', 'numpy==1.22.0', 'pandas<2,>=0.24', 'rdt>=0.2.7,<=0.4.0',