Updated the train method to be an InputArrayOfArrays, so the Python wrapper now generates the correct code (and it works).