Maximum recursion depth error

[stallam-unb]

UMAP Version: 0.5.3
Source: conda-forge
Python Version: 3.10.4
OS: Windows 11

I am trying to perform metric learning with UMAP. I train with supervised training data and using that on a new test data. However, I keep getting the following error:

RecursionError: Failed in nopython mode pipeline (step: convert to parfors)
maximum recursion depth exceeded while calling a Python object

There were similar issues reported here, for instance #99 , but the post indicates that it should be fixed. Adding small amounts of noise doesn't seem to help.

The training data is approx. 6k samples long and has 24 features, and the test data is approximately 50k samples long. Note that the error only happens with the test data. The training data can be transformed without any issues.