Fix inline docs for set_input