fix(setup): missing sklearn dependency