Ci-Dave's picture
removed sklearn as import
40b5a6f