Commented assert_raise_message and added assert_raises in order to be compliant with backwards compatibility requirements wrt sklearn