diff --git a/setup.py b/setup.py index 9060b734..4c4ad9a2 100644 --- a/setup.py +++ b/setup.py @@ -24,12 +24,12 @@ ] other_deps = [ 'pandas>=1.1,<1.5', - 'scikit-learn>=0.23,<1.1', + 'scikit-learn>=0.23,<1.3', ] pytorch_deps = [ 'opacus>=1.0.0,<1.1', 'Pillow>=8.0.1,<9.1.0', - 'scikit-learn>=0.23,<1.1', + 'scikit-learn>=0.23,<1.3', 'scipy>=1.5,<1.8', 'torch>=1.7,<1.11', 'torchsummary~=1.5.0',