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