diff --git a/setup.py b/setup.py index d973936..9ba1a1b 100644 --- a/setup.py +++ b/setup.py @@ -30,7 +30,7 @@ 'cvxopt', 'dill', 'h5py', - 'sklearn', + 'scikit-learn', 'grasp_nms' ] )