Replace probs. setup with TrainingAlgorithms as it broke