Update description for split_test_with_probability