Hello, I created a PR https://github.com/nccgroup/umap2/pull/41 in the original https://github.com/nccgroup/umap2 repo. That PR may be interesting for this project too. ~ Marius