diff --git a/pyproject.toml b/pyproject.toml index 9ffa4744..b165a184 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -42,11 +42,12 @@ dependencies = [ "pandas <3.0.0", "gradient-free-optimizers >=1.2.4, <2.0.0", "scikit-base <1.0.0", + "scikit-learn <1.7.0", ] [project.optional-dependencies] sklearn-integration = [ - "scikit-learn == 1.6.1", + "scikit-learn <1.7.0", ] build = [ "setuptools",