I messed this up. The order of `test_score` and `train_score` is wrong in the docstring.