Removed deprecated api usage in training_algorithms/tests