Pp.neighbors 'nopython mode pipeline error'

Hello,

I am having trouble running pp.neighbors. When i try to use it I get an error saying:

'LoweringError: Failed in nopython mode pipeline (step: nopython mode backend)
Storing i64 to ptr of i32 (‘dim’). FE type int32

File “…/…/…/opt/anaconda3/lib/python3.8/site-packages/umap/layouts.py”, line 52:
def rdist(x, y):

result = 0.0
dim = x.shape[0]
^

During: lowering “dim = static_getitem(value=$8load_attr.2, index=0, index_var=$const10.3, fn=)” at /Users/haimmoore/opt/anaconda3/lib/python3.8/site-packages/umap/layouts.py (52)’

Others reported this error too. The solution was to change versions of numba and llvmlite. I tried different combinations and pp.neighbors still gives the same errors. Right now I’m on Numpy 1.21.1 numba 0.53.1, and llvmlite0.36. I also tried llvmlite-0.37.0rc2, numba-0.54.0rc3, and numpy-1.20.3.

updating umap-learn to 0.5.1 fixed this issue