Minor tweak for OneVsRest.py