TST: this is just so the tests pass here.

This is substantially refactored on the tensor_predict branch and when the time
comes to merge this will all go away, but for now just making sure the tests
pass here.