[DOC] Fix an example with vectorize

The explicit vectorize import should be stated in this example, as it is intended to be a complete example.