PDIO-237 Fix TrainingTestSplit for non-rate action