To use UMAP you need to install umap-learn not umap.
So, in case you installed umap run the following commands to uninstall umap and install upam-learn instead:
pip uninstall umap
pip install umap-learn
And then in your python code make sure you are importing the module using:
import umap.umap_ as umap
Instead of
import umap