scikit-learn-contrib / project-template / 172

100%

Pull #33

travis-ci

Fix AssertionError when using TemplateTransformer

As-is, TemplateTransformer fails with the following code:

    from sklearn.utils.estimator_checks import check_estimator
    tpl = TemplateTransformer()
    check_estimator(tpl)

> AssertionError: Error message does not include the expected string: 'Reshape your data'. Observed error message: 'Shape of input is different from what was seen in `fit`'

Changing the message within ValueError to include the expected string fixes this issue.

1 of 1 new or added line in 1 file covered. (100.0%)

60 of 60 relevant lines covered (100.0%)

1.0 hits per line