diff --git a/setup.py b/setup.py index f759945fd..6582054b9 100644 --- a/setup.py +++ b/setup.py @@ -55,6 +55,7 @@ REQUIRED = [ "tornado", "joblib>=0.17.0", "ruamel.yaml>=0.16.12", + "scikit-learn>=0.22" ] # Numpy include