diff --git a/setup.py b/setup.py index c35dadb..494e07a 100644 --- a/setup.py +++ b/setup.py @@ -197,7 +197,7 @@ def build_extension(self, ext: CMakeExtension) -> None: 'pandas<=2.0', 'ruamel.yaml', 'psutil', - 'scikit-learn<=1.4.2' + 'scikit-learn<2' ], classifiers=[ 'Programming Language :: Python :: 3',