Add try/except block to handle development-version arguments to sklearn StratifiedKFold