Update layouts.py

Fix for #1006
lmcinnes · May 8, 2023 · c5b616c · c5b616c
1 parent 27a8912
commit c5b616c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/umap/layouts.py b/umap/layouts.py
@@ -35,6 +35,7 @@ def clip(val):
         "result": numba.types.float32,
         "diff": numba.types.float32,
         "dim": numba.types.intp,
+        "i": numba.types.intp,
     },
 )
 def rdist(x, y):