BUG: fix sklearn version requirement