Seems to work with latest sklearn, but needs testing and proper model evaluation