Updated mlagents learn cli test.