scratch that: keeping sklearn dependency