Python kernel unresponsive on using umap.UMAP().fit_transform()

[ac491]

The databricks python kernel goes unresponsive when running the code snippet:

umap.UMAP().fit_transform(np.random.randn(10_000, 20))

I am using python version 3.9, umap-learn == 0.5.4 and the following additional libraries are also installed:

scikit-learn==1.2.1
transformers==4.30.2
sentence-transformers==2.2.2
numpy==1.24.1
numba==0.57.0
typing-extensions==4.8.0

Thanks in advance!