diff --git a/pyproject.toml b/pyproject.toml index b6f47cfa..bf7af2a4 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -29,6 +29,7 @@ dependencies = [ "pandas>=1.1.3;python_version<'3.10'", "pandas>=1.3.4;python_version>='3.10' and python_version<'3.11'", "pandas>=1.5.0;python_version>='3.11'", + "scikit-learn>=0.24,<2;python_version<'3.10'", "scikit-learn>=1.1.3,<2;python_version>='3.10'", "torch>=1.8.0;python_version<'3.10'", "torch>=1.11.0;python_version>='3.10' and python_version<'3.11'",