Fix TrainAndPredictIrisModelUsingDirectInstantiationTest test from commit #428

PR #428 was merged July 12, many days after PR #468 which added a new test on the Iris dataset July 2. The changes in #428 were not applied to that new test, and with the merge that test is failing.