Documentation Figure title incorrect

Last figure in the below link the title would be "test digits" instead of train digits I guess. https://umap-learn.readthedocs.io/en/latest/supervised.html

Open the supervised learning documentation at the linked Read the Docs page and inspect the title of its final figure. Confirm whether the figure shows test digits rather than train digits, then update the documentation source and verify that the rendered title is correct.

Written by the indexing model from the issue text.